diff --git a/internal/querynodev2/services.go b/internal/querynodev2/services.go
index f2bffd8e7eb29a384a57a0944a072ed924e60ab2..66a7ee2f78ceefb774a0e932aec78f88d7ea07f9 100644
--- a/internal/querynodev2/services.go
+++ b/internal/querynodev2/services.go
@@ -821,6 +821,7 @@ func (node *QueryNode) Search(ctx context.Context, req *querypb.SearchRequest) (
 			SegmentIDs:      req.SegmentIDs,
 			FromShardLeader: req.FromShardLeader,
 			Scope:           req.Scope,
+			TotalChannelNum: req.TotalChannelNum,
 		}
 
 		runningGp.Go(func() error {
diff --git a/internal/querynodev2/services_test.go b/internal/querynodev2/services_test.go
index b6cbf70d1c707674e6e22574c105d94f616c65e3..860e371bad40e093380440e7b0c9e141bc6650a6 100644
--- a/internal/querynodev2/services_test.go
+++ b/internal/querynodev2/services_test.go
@@ -1091,6 +1091,7 @@ func (suite *ServiceSuite) TestSearch_Normal() {
 		Req:             creq,
 		FromShardLeader: false,
 		DmlChannels:     []string{suite.vchannel},
+		TotalChannelNum: 2,
 	}
 	suite.NoError(err)
 
@@ -1117,6 +1118,7 @@ func (suite *ServiceSuite) TestSearch_Concurrent() {
 				Req:             creq,
 				FromShardLeader: false,
 				DmlChannels:     []string{suite.vchannel},
+				TotalChannelNum: 2,
 			}
 			suite.NoError(err)
 			return suite.node.Search(ctx, req)
@@ -1142,6 +1144,7 @@ func (suite *ServiceSuite) TestSearch_Failed() {
 		Req:             creq,
 		FromShardLeader: false,
 		DmlChannels:     []string{suite.vchannel},
+		TotalChannelNum: 2,
 	}
 	suite.NoError(err)
 
@@ -1201,6 +1204,7 @@ func (suite *ServiceSuite) TestSearchSegments_Normal() {
 		Req:             creq,
 		FromShardLeader: true,
 		DmlChannels:     []string{suite.vchannel},
+		TotalChannelNum: 2,
 	}
 	suite.NoError(err)