diff --git a/.github/workflows/release-infra.yml b/.github/workflows/release-infra.yml
index dcc5e604..5fb15ace 100644
--- a/.github/workflows/release-infra.yml
+++ b/.github/workflows/release-infra.yml
@@ -41,7 +41,7 @@ jobs:
       - id: env-var
         run: |
           # prod env from main branch
-          if [[ "${{ github.ref_name }}" == refs/heads/main ]]; then
+          if [[ "${{ github.ref_name }}" == main ]]; then
             export ENVIRONMENT=prod
           else
             export ENVIRONMENT=$(echo ${{ github.ref_name }} | awk -F- '{ print $2 }')