diff --git a/.github/workflows/production-docker.yml b/.github/workflows/production-docker.yml
index 6b94711..d4e5c6b 100644
--- a/.github/workflows/production-docker.yml
+++ b/.github/workflows/production-docker.yml
@@ -22,7 +22,7 @@ jobs:
            whoami
            cd /root/nodbot
            git pull
-           git checkout main
+           git checkout $GITHUB_HEAD_REF
     - name: Build the Docker image
       run: |
            cd /root/nodbot