diff --git a/action.yml b/action.yml
index 445408d8..5bea4953 100644
--- a/action.yml
+++ b/action.yml
@@ -40,7 +40,7 @@ branding:
 runs:
   using: 'node20'
   pre: 'lib/cleanup/index.js'
-  pre-if: (! startsWith(runner.name, 'GitHub Actions'))
+  pre-if: (! env.AZURE_LOGIN_PRE_CLEANUP_OFF)
   main: 'lib/main/index.js'
   post: 'lib/cleanup/index.js'
-  post-if: (! startsWith(runner.name, 'GitHub Actions'))
+  post-if: (! env.AZURE_LOGIN_POST_CLEANUP_OFF)