diff --git a/.github/workflows/on_tag.yml b/.github/workflows/on_tag.yml
index 82de846..2f219e4 100644
--- a/.github/workflows/on_tag.yml
+++ b/.github/workflows/on_tag.yml
@@ -36,8 +36,6 @@ jobs:
     steps:
       - name: Check out code
         uses: actions/checkout@v4
-        with:
-          fetch-depth: 0
 
       - name: Docker Hub Login
         uses: docker/login-action@v3
@@ -50,3 +48,5 @@ jobs:
         with:
           push: true
           tags: ${{ env.CI_REGISTRY_IMAGE }}:${{ env.REGISTRY_IMAGE_TAG }}
+          build-args: |
+            REF_NAME=${{ env.REGISTRY_IMAGE_TAG }}
diff --git a/Dockerfile b/Dockerfile
index 6b13499..278497a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,7 @@
 FROM python:3.10-slim
 
+ARG REF_NAME=main  # Default to 'main' if REF_NAME is not passed
+
 RUN apt-get update && \
         DEBIAN_FRONTEND="noninteractive" TZ="Europe/Zurich" apt-get install -y tzdata && \
         apt-get install -y --no-install-recommends \
@@ -12,17 +14,17 @@ RUN apt-get update && \
 
 RUN apt-get -y install pip git vim
 
-#RUN pwd
-#RUN ls -a
-#RUN git clone https://github.com/BlueBrain/bbp-atlas-pipeline.git@ /pipeline
 RUN pwd
 RUN ls -a
-WORKDIR /pipeline
+RUN git clone https://github.com/BlueBrain/bbp-atlas-pipeline.git@${REF_NAME} /pipeline
 RUN pwd
 RUN ls -a
-COPY .. .
+WORKDIR /pipeline
 RUN pwd
 RUN ls -a
+#COPY .. .
+#RUN pwd
+#RUN ls -a
 
 # Regiodesics
 #RUN git clone https://bbpgitlab.epfl.ch/nse/archive/regiodesics  && \