diff --git a/Dockerfile-notebook b/Dockerfile-notebook
index 6e24567..7551b91 100644
--- a/Dockerfile-notebook
+++ b/Dockerfile-notebook
@@ -23,8 +23,7 @@ RUN cd /packages && \
     git checkout e16f637d079ef5d56321a240ef0547a50c37b708
 RUN cd /packages/axolotl/ && \
     MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install -U --no-build-isolation --no-cache-dir -e .[flash-attn,mamba-ssm,fused-dense-lib,optimizers,lion-pytorch,galore] && \
-    pip install --no-cache-dir -U -r /tmp/llm-finetune/notebook-requirements.txt && \
-    rm -rf /root/.cache/pip
+    pip install --no-cache-dir -U -r /tmp/llm-finetune/notebook-requirements.txt
 COPY plugins/axolotl_truefoundry /packages/axolotl_truefoundry
 RUN cd /packages/axolotl_truefoundry/ && \
     pip install --no-cache-dir -e .