diff --git a/docker/Dockerfile.cu121.mlc b/docker/Dockerfile.cu121.mlc
index 6cfc066..5ec7906 100644
--- a/docker/Dockerfile.cu121.mlc
+++ b/docker/Dockerfile.cu121.mlc
@@ -18,7 +18,7 @@ RUN bash <(curl -L micro.mamba.pm/install.sh) && source ~/.bashrc       && \
     micromamba activate python311                                       && \
     pip install --pre --force-reinstall -f https://mlc.ai/wheels           \
                 mlc-ai-nightly-cu121                                       \
-                mlc-chat-nightly-cu121
+                mlc-llm-nightly-cu121
 
 RUN git clone https://github.com/mlc-ai/mlc-llm.git /mlc_llm             && \
     echo "export PYTHONPATH=\"/mlc_llm/:$PYTHONPATH\"" >> ~/.bashrc