diff --git a/src/cli_demo.py b/src/cli_demo.py
index 05e8c8eb..fe6a0bc4 100644
--- a/src/cli_demo.py
+++ b/src/cli_demo.py
@@ -1,3 +1,4 @@
+import readline
 from llmtuner import ChatModel