diff --git a/cognee/tests/test_search_db.py b/cognee/tests/test_search_db.py
index 9eafb5c0c..8a99ebd8f 100644
--- a/cognee/tests/test_search_db.py
+++ b/cognee/tests/test_search_db.py
@@ -44,16 +44,16 @@ async def main():
 
     await cognee.cognify([dataset_name])
 
-    context_gk = await GraphCompletionRetriever().get_context(
+    context_gk, _ = await GraphCompletionRetriever().get_context(
         query="Next to which country is Germany located?"
     )
-    context_gk_cot = await GraphCompletionCotRetriever().get_context(
+    context_gk_cot, _ = await GraphCompletionCotRetriever().get_context(
         query="Next to which country is Germany located?"
     )
-    context_gk_ext = await GraphCompletionContextExtensionRetriever().get_context(
+    context_gk_ext, _ = await GraphCompletionContextExtensionRetriever().get_context(
         query="Next to which country is Germany located?"
     )
-    context_gk_sum = await GraphSummaryCompletionRetriever().get_context(
+    context_gk_sum, _ = await GraphSummaryCompletionRetriever().get_context(
         query="Next to which country is Germany located?"
     )