diff --git a/.github/workflows/dbt_udf_test.yml b/.github/workflows/dbt_udf_test.yml
index 2a190ee..0354867 100644
--- a/.github/workflows/dbt_udf_test.yml
+++ b/.github/workflows/dbt_udf_test.yml
@@ -58,5 +58,5 @@ jobs:
     secrets: inherit
     with:
       warehouse: ${{ vars.WAREHOUSE }}
-      environment: dev
+      environment: ${{ github.ref == 'refs/heads/main' && 'workflow_prod' || 'workflow_dev' }}
       command: dbt test --selector test_udfs --threads 24