diff --git a/agents-api/agents_api/models/docs/search_docs_by_embedding.py b/agents-api/agents_api/models/docs/search_docs_by_embedding.py
index 382238f99..382bff069 100644
--- a/agents-api/agents_api/models/docs/search_docs_by_embedding.py
+++ b/agents-api/agents_api/models/docs/search_docs_by_embedding.py
@@ -246,7 +246,7 @@ def search_docs_by_embedding(
 
         # Sort the results by distance to find the closest matches
         :sort -mmr_score
-        :limit {k*(3 if mmr_strength else 1)},   # Get more candidates for diversity
+        :limit {k*(3 if mmr_strength else 1)}   # Get more candidates for diversity
 
         :create _interim {{
             owner_type,