diff --git a/core/cli.py b/core/cli.py
index 93b7a7e1..54bca3b7 100644
--- a/core/cli.py
+++ b/core/cli.py
@@ -13,7 +13,7 @@ MODULES = {
     },
     "thread_manager": {
         "required": True,
-        "files": ["thread_manager.py", "ui.py"],
+        "files": ["thread_manager.py", "thread_viewer_ui.py"],
         "description": "Message thread management module - Manages conversation history and message flows"
     },
     "tool_system": {