diff --git a/apps/dataset/serializers/document_serializers.py b/apps/dataset/serializers/document_serializers.py
index e30e9f360b3..ad9199a52d1 100644
--- a/apps/dataset/serializers/document_serializers.py
+++ b/apps/dataset/serializers/document_serializers.py
@@ -1225,7 +1225,7 @@ def batch_generate_related(self, instance: Dict, with_valid=True):
                                         1),
             ).filter(task_type_status__in=state_list, document_id__in=document_id_list)
                                              .values('id'),
-                                             TaskType.EMBEDDING,
+                                             TaskType.GENERATE_PROBLEM,
                                              State.PENDING)
             ListenerManagement.get_aggregation_document_status_by_query_set(
                 QuerySet(Document).filter(id__in=document_id_list))()