diff --git a/actions/build_deploy_python_executable/action.yml b/actions/build_deploy_python_executable/action.yml
index 86094128..5fe88957 100644
--- a/actions/build_deploy_python_executable/action.yml
+++ b/actions/build_deploy_python_executable/action.yml
@@ -65,6 +65,7 @@ runs:
     - if: ${{ inputs.deploy == 'true' }}
       run: >
         cd $ACTION_REPO &&
+        INPUT_DEPLOYMENT=${{ inputs.deployment }}
         /usr/bin/python src/deploy_pex.py
         ${{ inputs.dagster_cloud_file }}
         --python-version=${{ inputs.python_version }}
diff --git a/src/deploy_pex.py b/src/deploy_pex.py
index c56d499e..207b4243 100755
--- a/src/deploy_pex.py
+++ b/src/deploy_pex.py
@@ -33,7 +33,7 @@ def main():
         project_dir = os.path.dirname(dagster_cloud_yaml)
         deployment_name = get_branch_deployment_name(project_dir)
     else:
-        # INPUT_DEPLOYMENT is automatically set by github to the `deployment:` input value, if provided
+        # INPUT_DEPLOYMENT is to the `deployment:` input value in action.yml
         deployment_name = os.getenv("INPUT_DEPLOYMENT", "prod")
         print(f"Deploying to a full deployment: {deployment_name}", flush=True)