diff --git a/Dockerfile b/Dockerfile
index ae06805..f27052d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,11 +2,13 @@ FROM node:18-bullseye-slim
 WORKDIR /app
 COPY package*.json ./
 ARG GIT_COMMIT
+ARG GIT_BRANCH
 RUN apt-get update
 RUN apt-get install -y git
 RUN rm -rf /var/lib/apt/lists/*
 RUN npm install
-RUN rm -rf .git
+ENV GIT_BRANCH=$GIT_BRANCH
+ENV GIT_COMMIT=$GIT_COMMIT
 RUN git clone -n https://github.com/wukko/cobalt.git --depth 1 && mv cobalt/.git ./ && rm -rf cobalt
 COPY . .
 EXPOSE 9000