diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py
index 5f56646ef..af8dfc186 100644
--- a/rag/svr/task_executor.py
+++ b/rag/svr/task_executor.py
@@ -193,7 +193,7 @@ async def collect():
     canceled = False
     if msg.get("doc_id", "") in [GRAPH_RAPTOR_FAKE_DOC_ID, CANVAS_DEBUG_DOC_ID]:
         task = msg
-        if task["task_type"] in ["graphrag", "raptor", "mindmap"]:
+        if task["task_type"] in PIPELINE_SPECIAL_PROGRESS_FREEZE_TASK_TYPES:
             task = TaskService.get_task(msg["id"], msg["doc_ids"])
             if task:
                 task["doc_id"] = msg["doc_id"]