diff --git a/.github/workflows/_deploy_lambda.yml b/.github/workflows/_deploy_lambda.yml
index b3ca4583..9bd686aa 100644
--- a/.github/workflows/_deploy_lambda.yml
+++ b/.github/workflows/_deploy_lambda.yml
@@ -27,6 +27,7 @@ on:
         required: false
         type: string
         default: 'false'
+        # can only be 'true' or 'false'
 
     secrets:
       AWS_ACCESS_KEY_ID:
@@ -92,6 +93,6 @@ jobs:
             -out=lambdaplan
 
       - name: Terraform Apply
-        if: inputs.terraform_apply == 'true' || inputs.stage == 'dev' || inputs.stage == 'main'
+        if: inputs.terraform_apply == 'true' || github.ref == 'refs/heads/dev' || github.ref == 'refs/heads/main'
         working-directory: ${{ inputs.lambda_path }}
         run: terraform apply -auto-approve lambdaplan