diff --git a/cognee/modules/search/methods/search.py b/cognee/modules/search/methods/search.py
index 405207114..0bc845a10 100644
--- a/cognee/modules/search/methods/search.py
+++ b/cognee/modules/search/methods/search.py
@@ -132,14 +132,31 @@ async def search(
             ],
         )
     else:
-        return [
-            SearchResult(
-                search_result=result,
-                dataset_id=datasets[min(index, len(datasets) - 1)].id if datasets else None,
-                dataset_name=datasets[min(index, len(datasets) - 1)].name if datasets else None,
-            )
-            for index, (result, _, datasets) in enumerate(search_results)
-        ]
+        # This is for maintaining backwards compatibility
+        if os.getenv("ENABLE_BACKEND_ACCESS_CONTROL", "false").lower() == "true":
+            return_value = {}
+            for search_result in search_results:
+                result, context, datasets = search_result
+                return_value[str(datasets[0].id)] = {
+                    "search_result": result,
+                    "dataset_id": str(datasets[0].id),
+                }
+            return return_value
+        else:
+            return_value = []
+            for search_result in search_results:
+                result, context, datasets = search_result
+                return_value.append(result)
+
+            return return_value
+        # return [
+        #     SearchResult(
+        #         search_result=result,
+        #         dataset_id=datasets[min(index, len(datasets) - 1)].id if datasets else None,
+        #         dataset_name=datasets[min(index, len(datasets) - 1)].name if datasets else None,
+        #     )
+        #     for index, (result, _, datasets) in enumerate(search_results)
+        # ]
 
 
 async def authorized_search(