diff --git a/Dockerfile b/Dockerfile
index 3e95d69..9680d68 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -18,9 +18,10 @@ 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 . && \
-    rm -rf /tmp/lmcache
+    rm -rf /tmp/lmcache
\ No newline at end of file