diff --git a/api/apps/services/dataset_api_service.py b/api/apps/services/dataset_api_service.py
index 5927b780ec..8d5f512a35 100644
--- a/api/apps/services/dataset_api_service.py
+++ b/api/apps/services/dataset_api_service.py
@@ -598,9 +598,8 @@ def aggregate_tags(dataset_ids: list[str], tenant_id: str):
 
     merged = {}
     for kb_tenant_id, kb_ids in dataset_ids_by_tenant.items():
-        for bucket in settings.retriever.all_tags(kb_tenant_id, kb_ids):
-            tag = bucket["value"]
-            merged[tag] = merged.get(tag, 0) + bucket["count"]
+        for tag, count in settings.retriever.all_tags(kb_tenant_id, kb_ids):
+            merged[tag] = merged.get(tag, 0) + count
 
     return True, [{"value": tag, "count": count} for tag, count in merged.items()]