diff --git a/internal/querycoord/task.go b/internal/querycoord/task.go
index 2518613ad4..9f38cb0345 100644
--- a/internal/querycoord/task.go
+++ b/internal/querycoord/task.go
@@ -1089,6 +1089,11 @@ func (lst *loadSegmentTask) reschedule(ctx context.Context) ([]task, error) {
 			Schema:       lst.Schema,
 			SourceNodeID: lst.SourceNodeID,
 			CollectionID: lst.CollectionID,
+			LoadMeta: &querypb.LoadMetaInfo{
+				LoadType:     lst.GetLoadMeta().GetLoadType(),
+				CollectionID: lst.GetCollectionID(),
+				PartitionIDs: lst.GetLoadMeta().GetPartitionIDs(),
+			},
 		}
 		loadSegmentReqs = append(loadSegmentReqs, req)
 	}
@@ -1263,6 +1268,11 @@ func (wdt *watchDmChannelTask) reschedule(ctx context.Context) ([]task, error) {
 			Infos:        []*datapb.VchannelInfo{info},
 			Schema:       wdt.Schema,
 			ExcludeInfos: wdt.ExcludeInfos,
+			LoadMeta: &querypb.LoadMetaInfo{
+				LoadType:     wdt.GetLoadMeta().GetLoadType(),
+				CollectionID: collectionID,
+				PartitionIDs: wdt.GetLoadMeta().GetPartitionIDs(),
+			},
 		}
 		watchDmChannelReqs = append(watchDmChannelReqs, req)
 	}
diff --git a/internal/querynode/task.go b/internal/querynode/task.go
index da3271574b..301423239f 100644
--- a/internal/querynode/task.go
+++ b/internal/querynode/task.go
@@ -631,7 +631,7 @@ func (l *loadSegmentsTask) Execute(ctx context.Context) error {
 	var err error
 
 	// init meta
-	collectionID := l.req.GetLoadMeta().GetCollectionID()
+	collectionID := l.req.GetCollectionID()
 	l.node.historical.replica.addCollection(collectionID, l.req.GetSchema())
 	l.node.streaming.replica.addCollection(collectionID, l.req.GetSchema())
 	for _, partitionID := range l.req.GetLoadMeta().GetPartitionIDs() {