diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml
index 67f6537..8dd3831 100644
--- a/.github/workflows/release.yaml
+++ b/.github/workflows/release.yaml
@@ -43,10 +43,15 @@ jobs:
           git push origin main
 
       # Switch back to the branch that triggered the workflow
-      - name: Checkout original branch
+      - name: Checkout original branch (if not main)
         run: |
-          git checkout ${{ env.BRANCH_NAME }}
-          git pull origin ${{ env.BRANCH_NAME }}
+            if [ "${BRANCH_NAME}" != "main" ]; then
+              git checkout "${BRANCH_NAME}"
+              git pull origin "${BRANCH_NAME}"
+            else
+              echo "Already on main branch; no need to checkout."
+            fi
+
 
       - name: npm build
         run: |