diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
index 9f97e022..fec5b7e1 100644
--- a/.github/workflows/docker-publish.yml
+++ b/.github/workflows/docker-publish.yml
@@ -70,6 +70,7 @@ jobs:
           push: ${{ github.event_name != 'pull_request' }}
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
+          build-args: BUILD_BRANCH=${{ github.ref_name }}
 
       # Sign the resulting Docker image digest except on PRs.
       # This will only write to the public Rekor transparency log when the Docker
diff --git a/Dockerfile b/Dockerfile
index 841dae16..77a32d27 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -7,8 +7,9 @@ WORKDIR /app
 # RUN tarball_url=$(curl -s https://api.github.com/repos/koodo-reader/koodo-reader/releases/latest | jq -r ".tarball_url") \
 #     && wget -qO- $tarball_url \
 #     | tar xvfz - --strip 1
+ARG BUILD_BRANCH=master
 RUN git clone https://github.com/koodo-reader/koodo-reader.git . && \
-    git checkout master
+    git checkout ${BUILD_BRANCH}
 
 ### --network-timeout 1000000 as a workaround for slow devices
 ### when the package being installed is too large, Yarn assumes it's a network problem and throws an error