diff --git a/.github/workflows/dbt_run_template.yml b/.github/workflows/dbt_run_template.yml
index a654abd1..017a210f 100644
--- a/.github/workflows/dbt_run_template.yml
+++ b/.github/workflows/dbt_run_template.yml
@@ -58,7 +58,7 @@ jobs:
           if [[ "${{ inputs.target }}" == *"prod"* ]]; then
             echo "DATABASE=$PROFILE_NAME" >> $GITHUB_ENV
             echo "ROLE=DBT_CLOUD_$PROFILE_NAME" >> $GITHUB_ENV
-            echo "WAREHOUSE=DBT_CLOUD" >> $GITHUB_ENV
+            echo "WAREHOUSE=${{ inputs.warehouse }}" >> $GITHUB_ENV
           elif [[ "${{ inputs.target }}" == *"test"* ]]; then
             echo "DATABASE=$PROFILE_NAME" >> $GITHUB_ENV
             echo "ROLE=DBT_CLOUD_$PROFILE_NAME" >> $GITHUB_ENV