diff --git a/mlc/mlc.py b/mlc/mlc.py
index 2dc2628..32d9175 100644
--- a/mlc/mlc.py
+++ b/mlc/mlc.py
@@ -1,5 +1,5 @@
-from mlc_chat import ChatModule, ChatConfig
-from mlc_chat.callback import StreamToStdout
+from mlc_llm import ChatModule, ChatConfig
+from mlc_llm.callback import StreamToStdout
 from transformers import AutoTokenizer
 import time
 import sys