diff --git a/cognee/tests/unit/modules/search/search_methods_test.py b/cognee/tests/unit/modules/search/search_methods_test.py
index 8645d965a..46995d087 100644
--- a/cognee/tests/unit/modules/search/search_methods_test.py
+++ b/cognee/tests/unit/modules/search/search_methods_test.py
@@ -66,6 +66,7 @@ async def test_search(
         node_type=None,
         node_name=None,
         save_interaction=False,
+        last_k=None,
     )
 
     # Verify result logging