diff --git a/.github/workflows/contextive-docs.yml b/.github/workflows/contextive-docs.yml
index 2ee1d47..1facc28 100644
--- a/.github/workflows/contextive-docs.yml
+++ b/.github/workflows/contextive-docs.yml
@@ -5,6 +5,8 @@ on:
       - 'docs/web/**'
       - '.github/workflows/contextive-docs.yml'
 
+env:
+  BRANCH_NAME: ${{ github.ref_name }}
 
 jobs:
   docs:
@@ -28,7 +30,10 @@ jobs:
 
       - run: npm run build
 
-      - run: npm run deploy
+      - run: |
+          prod_flag=""
+          if [ "$BRANCH_NAME" = "main" ]; then prod_flag="--prod"; fi
+          npm run deploy -- $prod_flag
         env:
           NETLIFY_SITE_ID: ${{ secrets.NETLIFY_SITE_ID }}
           NETLIFY_AUTH: ${{ secrets.NETLIFY_AUTH }} 
\ No newline at end of file