diff --git a/Dockerfile b/Dockerfile
index f883caf..7b6697e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,8 +9,9 @@ RUN apt-get update && apt-get install -y git \
     libcurand-dev-13-0 \
     libcufft-dev-13-0 \
     libnvjitlink-dev-13-0 && \
-    git clone --depth 1 https://github.com/LMCache/LMCache.git /tmp/lmcache && \
+    git clone https://github.com/neuralwatt/LMCache.git /tmp/lmcache && \
     cd /tmp/lmcache && \
+    git checkout fix/mla-multi-group-kv-cache && \
     CUDA_HOME=/usr/local/cuda \
     TORCH_CUDA_ARCH_LIST="10.0" \
     pip install --no-cache-dir --no-build-isolation . && \