diff --git a/.github/workflows/lambda_flow.yml b/.github/workflows/lambda_flow.yml
index 7499631..7648521 100644
--- a/.github/workflows/lambda_flow.yml
+++ b/.github/workflows/lambda_flow.yml
@@ -94,9 +94,11 @@ jobs:
           else
           deploy_environment=qa
           fi
-          elif [ "${{ github.event_name }}" == "push" ]; then
-          # If the PR is merged, deploy to the prod environment
+          elif [ "${{ github.ref }}" == "refs/heads/main" ]; then
+          # Push to main or workflow_dispatch on main → deploy to prod
               deploy_environment=prod
+          else
+              deploy_environment=default
           fi
           echo "deploy_environment:$deploy_environment"
           echo "deploy_environment=$deploy_environment" >> $GITHUB_OUTPUT