diff --git a/server/utils/agents/aibitat/plugins/memory.js b/server/utils/agents/aibitat/plugins/memory.js
index d31a264d18dd022a09262b43f2feee274f3d2bac..4f43d0ec460beff523025752285758a38e23890c 100644
--- a/server/utils/agents/aibitat/plugins/memory.js
+++ b/server/utils/agents/aibitat/plugins/memory.js
@@ -94,6 +94,7 @@ const memory = {
                   namespace: workspace.slug,
                   input: query,
                   LLMConnector,
+                  topN: workspace?.topN ?? 4,
                 });
 
               if (contextTexts.length === 0) {