diff --git a/cognee/api/client.py b/cognee/api/client.py
index 89292cd39..74cefcb8e 100644
--- a/cognee/api/client.py
+++ b/cognee/api/client.py
@@ -328,10 +328,10 @@ def start_api_server(host: str = "0.0.0.0", port: int = 8000):
     try:
         logger.info("Starting server at %s:%s", host, port)
 
-        # import asyncio
-        # from cognee.modules.data.deletion import prune_system, prune_data
-        # asyncio.run(prune_data())
-        # asyncio.run(prune_system(metadata = True))
+        import asyncio
+        from cognee.modules.data.deletion import prune_system, prune_data
+        asyncio.run(prune_data())
+        asyncio.run(prune_system(metadata = True))
 
         uvicorn.run(app, host = host, port = port)
     except Exception as e:
diff --git a/cognee/api/v1/search/search.py b/cognee/api/v1/search/search.py
index 551bb270e..7bd363e7d 100644
--- a/cognee/api/v1/search/search.py
+++ b/cognee/api/v1/search/search.py
@@ -55,9 +55,10 @@ async def search(search_type: str, params: Dict[str, Any], user: User = None) ->
     filtered_search_results = []
 
     for search_result in search_results:
-        result_document_id = UUID(search_result["document_id"]) if "document_id" in search_result else None
+        document_id = search_result["document_id"] if "document_id" in search_result else None
+        document_id = UUID(document_id) if type(document_id) == str else document_id
 
-        if result_document_id is None or result_document_id in own_document_ids:
+        if document_id is None or document_id in own_document_ids:
             filtered_search_results.append(search_result)
 
     return filtered_search_results