diff --git a/docker/Dockerfile b/docker/Dockerfile
index c122f1de1..3d3373eb1 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -12,5 +12,6 @@ WORKDIR /workspace
 RUN python3 -m pip install -U --no-cache-dir pydantic
 
 ARG REF=main
-RUN git clone https://github.com/winglian/axolotl && cd axolotl && git checkout $REF && \
-    pip install -e .[int4]
+RUN mkdir axolotl
+COPY . axolotl/
+RUN pip install -e .[int4]