diff --git a/.github/workflows/setup_docker.yml b/.github/workflows/setup_docker.yml
index 0dfcdd09e..458a4567e 100644
--- a/.github/workflows/setup_docker.yml
+++ b/.github/workflows/setup_docker.yml
@@ -12,12 +12,14 @@ jobs:
     steps:
       - uses: actions/checkout@v4
         with:
-          ref: ${{ github.head_ref }}
+          ref: ${{ github.head_ref || github.ref_name }} 
       - name: Create virtual environment
         run: pyenv virtualenv "$(git rev-parse --short "$GITHUB_SHA")"
   build:
     runs-on: self-hosted
     steps:
+      - name: update tag
+        run: make update_tag CURR_BRANCH=${{ github.head_ref || github.ref_name }} 
       - name: Stop Server Docker
         run: make stop_server_docker
       - name: Generate Config