diff --git a/docker/Dockerfile b/docker/Dockerfile
index 3d3373eb1..d87ac8f9d 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -14,4 +14,5 @@ RUN python3 -m pip install -U --no-cache-dir pydantic
 ARG REF=main
 RUN mkdir axolotl
 COPY . axolotl/
-RUN pip install -e .[int4]
+RUN cd axolotl && \
+    pip install -e .[int4]