diff --git a/.github/workflows/dbt_udf_test.yml b/.github/workflows/dbt_udf_test.yml
index 8ea8447..e201545 100644
--- a/.github/workflows/dbt_udf_test.yml
+++ b/.github/workflows/dbt_udf_test.yml
@@ -40,7 +40,7 @@ jobs:
     secrets: inherit
     with:
       warehouse: ${{ vars.WAREHOUSE }}
-      environment: prod
+      environment: ${{ github.ref == 'refs/heads/main' && 'workflow_prod' || 'workflow_dev' }}
       command: dbt test --selector test_udfs --threads 24
 
   dispatched: