diff --git a/agents-api/tests/test_docs_queries.py b/agents-api/tests/test_docs_queries.py
index b6b23438a..6690badfd 100644
--- a/agents-api/tests/test_docs_queries.py
+++ b/agents-api/tests/test_docs_queries.py
@@ -272,7 +272,7 @@ async def _(
     result = await search_docs_by_embedding(
         developer_id=developer.id,
         owners=[("agent", agent.id)],
-        query_embedding=query_embedding,
+        embedding=query_embedding,
         k=3,  # Add k parameter
         metadata_filter={"test": "test"},  # Add metadata filter
         connection_pool=pool,