diff --git a/cognee/__init__.py b/cognee/__init__.py
index 1720f6f8e..e1533b054 100644
--- a/cognee/__init__.py
+++ b/cognee/__init__.py
@@ -1,4 +1,4 @@
 from .api.v1.add.add import add
 from .api.v1.cognify.cognify import cognify
 from .api.v1.list_datasets.list_datasets import list_datasets
-from .api.v1.search import search
+from .api.v1.search.search import search
diff --git a/cognee/api/v1/search/search.py b/cognee/api/v1/search/search.py
index 7591f10f7..f9d787b94 100644
--- a/cognee/api/v1/search/search.py
+++ b/cognee/api/v1/search/search.py
@@ -8,7 +8,7 @@ from cognee.modules.search.vector.search_similarity import search_similarity
 from cognee.modules.search.graph.search_categories import search_categories
 from cognee.modules.search.graph.search_neighbour import search_neighbour
 from cognee.shared.data_models import GraphDBType
-
+import asyncio
 
 class SearchType(Enum):
     ADJACENT = auto()
@@ -48,7 +48,7 @@ async def search(graph, query_params: Dict[SearchType, Dict[str, Any]]) -> List:
     return results
 
 if __name__ == "__main__":
-    import asyncio
+