diff --git a/cognee/cli/tui/app.py b/cognee/cli/tui/app.py
index 4d8ac5949..ae7452306 100644
--- a/cognee/cli/tui/app.py
+++ b/cognee/cli/tui/app.py
@@ -118,7 +118,7 @@ class HelpScreen(Screen):
             self.app.pop_screen()
 
 
-def run_tui():
+def run_tui(mouse: bool = True):
     """Entry point to run the TUI application