diff --git a/cognee/cli/tui/cognify_screen.py b/cognee/cli/tui/cognify_screen.py
index 844cb345b..1108d6a7d 100644
--- a/cognee/cli/tui/cognify_screen.py
+++ b/cognee/cli/tui/cognify_screen.py
@@ -1,6 +1,4 @@
 import asyncio
-
-import cognee
 from cognee.modules.chunking.TextChunker import TextChunker
 from textual.app import ComposeResult
 from textual.widgets import Input, Label, Static, Checkbox, RadioSet, RadioButton
@@ -138,7 +136,7 @@ class CognifyTUIScreen(BaseTUIScreen):
             
             # Prepare datasets parameter
             datasets = [dataset_name] if dataset_name else None
-            
+            import cognee
             await cognee.cognify(
                 datasets=datasets,
                 chunker=chunker_class,
diff --git a/cognee/cli/tui/search_screen.py b/cognee/cli/tui/search_screen.py
index e04165b60..831178842 100644
--- a/cognee/cli/tui/search_screen.py
+++ b/cognee/cli/tui/search_screen.py
@@ -1,12 +1,9 @@
 import asyncio
-
-import cognee
 from cognee.modules.search.types import SearchType
 from textual.app import ComposeResult
 from textual.widgets import Input, Label, Static, Select
 from textual.containers import Container, Vertical, ScrollableContainer
 from textual.binding import Binding
-
 from cognee.cli.tui.base_screen import BaseTUIScreen
 
 
@@ -141,7 +138,7 @@ class SearchTUIScreen(BaseTUIScreen):
         try:
             # Convert string to SearchType enum
             search_type = SearchType[query_type]
-
+            import cognee
             # Perform search
             results = await cognee.search(
                 query_text=query_text,