diff --git a/.github/workflows/update-code-version.yml b/.github/workflows/update-code-version.yml
index 1e034097..bc0087da 100644
--- a/.github/workflows/update-code-version.yml
+++ b/.github/workflows/update-code-version.yml
@@ -19,7 +19,7 @@ jobs:
         run: echo "VERSION=${GITHUB_REF#refs/tags/v}" >> $GITHUB_ENV
 
       - name: Select main branch
-        run: git checkout main
+        run: git checkout origin/main
 
       - name: Update version in pyproject.toml
         run: |