diff --git a/internal/querycoord/task_scheduler.go b/internal/querycoord/task_scheduler.go
index 7eb98ad017..8c44ed2010 100644
--- a/internal/querycoord/task_scheduler.go
+++ b/internal/querycoord/task_scheduler.go
@@ -137,6 +137,7 @@ type TaskScheduler struct {
 	cancel context.CancelFunc
 }
 
+// NewTaskScheduler reloads tasks from kv and returns a new taskScheduler
 func NewTaskScheduler(ctx context.Context, meta Meta, cluster Cluster, kv *etcdkv.EtcdKV, rootCoord types.RootCoord, dataCoord types.DataCoord) (*TaskScheduler, error) {
 	ctx1, cancel := context.WithCancel(ctx)
 	taskChan := make(chan task, 1024)