diff --git a/agent/tools/retrieval.py b/agent/tools/retrieval.py
index 6c7ca8695..912a5c348 100644
--- a/agent/tools/retrieval.py
+++ b/agent/tools/retrieval.py
@@ -196,7 +196,7 @@ class Retrieval(ToolBase, ABC):
                 self._param.similarity_threshold,
                 1 - self._param.keywords_similarity_weight,
                 doc_ids=doc_ids,
-                aggs=False,
+                aggs=True,
                 rerank_mdl=rerank_mdl,
                 rank_feature=label_question(query, kbs),
             )