diff --git a/plugin/task_manager/cluster_comparison/cluster_comparison.go b/plugin/task_manager/cluster_comparison/cluster_comparison.go
index 982419f2..72a07722 100644
--- a/plugin/task_manager/cluster_comparison/cluster_comparison.go
+++ b/plugin/task_manager/cluster_comparison/cluster_comparison.go
@@ -97,7 +97,9 @@ func (p *processor) splitMajorTask(taskItem *task.Task) error {
 
 	var sourceTotalDocs int64
 	var targetTotalDocs int64
-	ctx := context.Background()
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Second*60))
+	defer cancel()
 
 	for i, index := range clusterComparisonTask.Indices {
 		sourceDump := migration_model.IndexComparisonDumpConfig{
diff --git a/plugin/task_manager/cluster_migration/cluster_migration.go b/plugin/task_manager/cluster_migration/cluster_migration.go
index 4424c0cd..27d9c757 100644
--- a/plugin/task_manager/cluster_migration/cluster_migration.go
+++ b/plugin/task_manager/cluster_migration/cluster_migration.go
@@ -96,7 +96,9 @@ func (p *processor) splitMajorMigrationTask(taskItem *task.Task) error {
 	pids = append(pids, taskItem.ID)
 
 	var totalDocs int64
-	ctx := context.Background()
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Duration(time.Second*60))
+	defer cancel()
 
 	for i, index := range clusterMigrationTask.Indices {
 		source := migration_model.IndexMigrationSourceConfig{