diff --git a/.github/workflows/terraform-deploy-infra.yml b/.github/workflows/terraform-deploy-infra.yml
index c1c3790..b5a3149 100644
--- a/.github/workflows/terraform-deploy-infra.yml
+++ b/.github/workflows/terraform-deploy-infra.yml
@@ -12,8 +12,7 @@ jobs:
         uses: actions/checkout@v4
       - name: Set env.BRANCH
         run: |
-          BRANCH=develop
-          #BRANCH=$(echo $GITHUB_REF | cut -d'/' -f 3)
+          BRANCH=$(echo $GITHUB_REF | cut -d'/' -f 3)
           case ${BRANCH} in
             develop)
               CF_SPACE="dev"
@@ -21,6 +20,9 @@ jobs:
             main)
               CF_SPACE="prod"
               ;;
+            stage)
+              CF_SPACE="stage"
+              ;;
           esac
           echo "BRANCH=${BRANCH}" >> $GITHUB_ENV
           echo "CF_SPACE=${CF_SPACE}" >> $GITHUB_ENV