diff --git a/runtime/datamate-python/app/module/annotation/service/mapping.py b/runtime/datamate-python/app/module/annotation/service/mapping.py
index 5884ad6..cad7362 100644
--- a/runtime/datamate-python/app/module/annotation/service/mapping.py
+++ b/runtime/datamate-python/app/module/annotation/service/mapping.py
@@ -49,11 +49,11 @@ class DatasetMappingService:
         Returns:
             (total_count, annotated_count) 元组
         """
-        # 获取数据集总数据量（只统计 ACTIVE 状态的文件）
+        # 获取数据集总数据量（统计 ACTIVE 和 COMPLETED 状态的文件）
         total_result = await self.db.execute(
             select(func.count()).select_from(DatasetFiles).where(
                 DatasetFiles.dataset_id == dataset_id,
-                DatasetFiles.status == "ACTIVE",
+                DatasetFiles.status.in_(["ACTIVE", "COMPLETED"]),
             )
         )
         total_count = int(total_result.scalar() or 0)