diff --git a/docker/Dockerfile b/docker/Dockerfile
index a47d856c8..7f5f9df32 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -11,7 +11,6 @@ WORKDIR /workspace
 # The base image ships with `pydantic==1.8.2` which is not working
 RUN python3 -m pip install -U --no-cache-dir pydantic
 
-ARG REF=main
 RUN mkdir axolotl
 COPY . axolotl/
 RUN cd axolotl && \
diff --git a/docker/Dockerfile-runpod b/docker/Dockerfile-runpod
index 019cff74c..cf47b3670 100644
--- a/docker/Dockerfile-runpod
+++ b/docker/Dockerfile-runpod
@@ -5,7 +5,7 @@ RUN apt install --yes --no-install-recommends openssh-server tmux && \
     mkdir -p ~/.ssh && \
     chmod 700 ~/.ssh && \
     echo -e "\n[[ -z \"\$TMUX\"  ]] && { tmux attach-session -t ssh_tmux || tmux new-session -s ssh_tmux; exit; }" >> ~/.bashrc && \
-    chmod +x /workdir/axolotl/scripts/runpod-entrypoint.sh
+    chmod +x /workspace/axolotl/scripts/runpod-entrypoint.sh
 
-ENTRYPOINT ["/workdir/axolotl/scripts/runpod-entrypoint.sh"]
+ENTRYPOINT ["/workspace/axolotl/scripts/runpod-entrypoint.sh"]
 CMD ["sleep", "infinity"]