diff --git a/.github/workflows/dbt_integration_test.yml b/.github/workflows/dbt_integration_test.yml
index 7bdcc42..1ee62cd 100644
--- a/.github/workflows/dbt_integration_test.yml
+++ b/.github/workflows/dbt_integration_test.yml
@@ -12,6 +12,6 @@ jobs:
     with:
       command: >
         dbt test --selector 'integration_tests'
-      environment:  ${{ github.ref == 'refs/heads/main' && 'workflow_prod' || 'workflow_dev' }}
+      environment: ${{ inputs.environment }}
       warehouse: ${{ vars.WAREHOUSE }}
     secrets: inherit
\ No newline at end of file