diff --git a/internal/datacoord/index_meta.go b/internal/datacoord/index_meta.go
index d0f3ef22dc..0a33c18764 100644
--- a/internal/datacoord/index_meta.go
+++ b/internal/datacoord/index_meta.go
@@ -623,26 +623,24 @@ func (m *indexMeta) IsUnIndexedSegment(collectionID UniqueID, segID UniqueID) bo
 	return false
 }
 
-func (m *indexMeta) getSegmentIndexes(segID UniqueID) map[UniqueID]*model.SegmentIndex {
+func (m *indexMeta) GetSegmentsIndexes(collectionID UniqueID, segIDs []UniqueID) map[int64]map[UniqueID]*model.SegmentIndex {
 	m.RLock()
 	defer m.RUnlock()
-
-	ret := make(map[UniqueID]*model.SegmentIndex, 0)
-	segIndexInfos, ok := m.segmentIndexes[segID]
-	if !ok || len(segIndexInfos) == 0 {
-		return ret
+	segmentsIndexes := make(map[int64]map[UniqueID]*model.SegmentIndex)
+	for _, segmentID := range segIDs {
+		segmentsIndexes[segmentID] = m.getSegmentIndexes(collectionID, segmentID)
 	}
-
-	for _, segIdx := range segIndexInfos {
-		ret[segIdx.IndexID] = model.CloneSegmentIndex(segIdx)
-	}
-	return ret
+	return segmentsIndexes
 }
 
 func (m *indexMeta) GetSegmentIndexes(collectionID UniqueID, segID UniqueID) map[UniqueID]*model.SegmentIndex {
 	m.RLock()
 	defer m.RUnlock()
+	return m.getSegmentIndexes(collectionID, segID)
+}
 
+// Note: thread-unsafe, don't call it outside indexMeta
+func (m *indexMeta) getSegmentIndexes(collectionID UniqueID, segID UniqueID) map[UniqueID]*model.SegmentIndex {
 	ret := make(map[UniqueID]*model.SegmentIndex, 0)
 	segIndexInfos, ok := m.segmentIndexes[segID]
 	if !ok || len(segIndexInfos) == 0 {
diff --git a/internal/datacoord/index_meta_test.go b/internal/datacoord/index_meta_test.go
index 5fce622421..991100a240 100644
--- a/internal/datacoord/index_meta_test.go
+++ b/internal/datacoord/index_meta_test.go
@@ -870,12 +870,12 @@ func TestMeta_GetSegmentIndexes(t *testing.T) {
 	m := createMeta(catalog, withIndexMeta(createIndexMeta(catalog)))
 
 	t.Run("success", func(t *testing.T) {
-		segIndexes := m.indexMeta.getSegmentIndexes(segID)
+		segIndexes := m.indexMeta.GetSegmentIndexes(collID, segID)
 		assert.Equal(t, 1, len(segIndexes))
 	})
 
 	t.Run("segment not exist", func(t *testing.T) {
-		segIndexes := m.indexMeta.getSegmentIndexes(segID + 100)
+		segIndexes := m.indexMeta.GetSegmentIndexes(collID, segID+100)
 		assert.Equal(t, 0, len(segIndexes))
 	})
 
diff --git a/internal/datacoord/index_service.go b/internal/datacoord/index_service.go
index c31559fbce..d7e77e6611 100644
--- a/internal/datacoord/index_service.go
+++ b/internal/datacoord/index_service.go
@@ -862,8 +862,9 @@ func (s *Server) GetIndexInfos(ctx context.Context, req *indexpb.GetIndexInfoReq
 		SegmentInfo: map[int64]*indexpb.SegmentInfo{},
 	}
 
+	segmentsIndexes := s.meta.indexMeta.GetSegmentsIndexes(req.GetCollectionID(), req.GetSegmentIDs())
 	for _, segID := range req.GetSegmentIDs() {
-		segIdxes := s.meta.indexMeta.GetSegmentIndexes(req.GetCollectionID(), segID)
+		segIdxes := segmentsIndexes[segID]
 		ret.SegmentInfo[segID] = &indexpb.SegmentInfo{
 			CollectionID: req.GetCollectionID(),
 			SegmentID:    segID,
diff --git a/internal/datacoord/task_scheduler.go b/internal/datacoord/task_scheduler.go
index 5b26886630..08fe7d57dc 100644
--- a/internal/datacoord/task_scheduler.go
+++ b/internal/datacoord/task_scheduler.go
@@ -111,7 +111,7 @@ func (s *taskScheduler) Stop() {
 func (s *taskScheduler) reloadFromMeta() {
 	segments := s.meta.GetAllSegmentsUnsafe()
 	for _, segment := range segments {
-		for _, segIndex := range s.meta.indexMeta.getSegmentIndexes(segment.ID) {
+		for _, segIndex := range s.meta.indexMeta.GetSegmentIndexes(segment.GetCollectionID(), segment.ID) {
 			if segIndex.IsDeleted {
 				continue
 			}
diff --git a/internal/querycoordv2/checkers/index_checker.go b/internal/querycoordv2/checkers/index_checker.go
index 4297df396a..2afabc6bf3 100644
--- a/internal/querycoordv2/checkers/index_checker.go
+++ b/internal/querycoordv2/checkers/index_checker.go
@@ -35,6 +35,8 @@ import (
 	"github.com/milvus-io/milvus/pkg/util/typeutil"
 )
 
+const MaxSegmentNumPerGetIndexInfoRPC = 1024
+
 var _ Checker = (*IndexChecker)(nil)
 
 // IndexChecker perform segment index check.
@@ -132,18 +134,21 @@ func (c *IndexChecker) checkReplica(ctx context.Context, collection *meta.Collec
 	}
 
 	segmentsToUpdate := typeutil.NewSet[int64]()
-	for segment, fields := range targets {
-		missingFields := typeutil.NewSet(fields...)
-		infos, err := c.broker.GetIndexInfo(ctx, collection.GetCollectionID(), segment)
+	for _, segmentIDs := range lo.Chunk(lo.Keys(idSegments), MaxSegmentNumPerGetIndexInfoRPC) {
+		segmentIndexInfos, err := c.broker.GetIndexInfo(ctx, collection.GetCollectionID(), segmentIDs...)
 		if err != nil {
-			log.Warn("failed to get indexInfo for segment", zap.Int64("segmentID", segment), zap.Error(err))
+			log.Warn("failed to get indexInfo for segments", zap.Int64s("segmentIDs", segmentIDs), zap.Error(err))
 			continue
 		}
-		for _, info := range infos {
-			if missingFields.Contain(info.GetFieldID()) &&
-				info.GetEnableIndex() &&
-				len(info.GetIndexFilePaths()) > 0 {
-				segmentsToUpdate.Insert(segment)
+		for segmentID, segmentIndexInfo := range segmentIndexInfos {
+			fields := targets[segmentID]
+			missingFields := typeutil.NewSet(fields...)
+			for _, fieldIndexInfo := range segmentIndexInfo {
+				if missingFields.Contain(fieldIndexInfo.GetFieldID()) &&
+					fieldIndexInfo.GetEnableIndex() &&
+					len(fieldIndexInfo.GetIndexFilePaths()) > 0 {
+					segmentsToUpdate.Insert(segmentID)
+				}
 			}
 		}
 	}
diff --git a/internal/querycoordv2/checkers/index_checker_test.go b/internal/querycoordv2/checkers/index_checker_test.go
index ae4d04f868..aaa8f341a8 100644
--- a/internal/querycoordv2/checkers/index_checker_test.go
+++ b/internal/querycoordv2/checkers/index_checker_test.go
@@ -116,14 +116,14 @@ func (suite *IndexCheckerSuite) TestLoadIndex() {
 
 	// broker
 	suite.broker.EXPECT().GetIndexInfo(mock.Anything, int64(1), int64(2)).
-		Return([]*querypb.FieldIndexInfo{
+		Return(map[int64][]*querypb.FieldIndexInfo{2: {
 			{
 				FieldID:        101,
 				IndexID:        1000,
 				EnableIndex:    true,
 				IndexFilePaths: []string{"index"},
 			},
-		}, nil)
+		}}, nil)
 
 	suite.broker.EXPECT().ListIndexes(mock.Anything, int64(1)).Return([]*indexpb.IndexInfo{
 		{
@@ -180,28 +180,28 @@ func (suite *IndexCheckerSuite) TestIndexInfoNotMatch() {
 	checker.dist.SegmentDistManager.Update(1, utils.CreateTestSegment(1, 1, 3, 1, 1, "test-insert-channel"))
 
 	// broker
-	suite.broker.EXPECT().GetIndexInfo(mock.Anything, int64(1), mock.AnythingOfType("int64")).Call.
-		Return(func(ctx context.Context, collectionID, segmentID int64) []*querypb.FieldIndexInfo {
-			if segmentID == 2 {
-				return []*querypb.FieldIndexInfo{
+	suite.broker.EXPECT().GetIndexInfo(mock.Anything, int64(1), mock.AnythingOfType("int64")).
+		RunAndReturn(func(ctx context.Context, collectionID int64, segmentIDs ...int64) (map[int64][]*querypb.FieldIndexInfo, error) {
+			if segmentIDs[0] == 2 {
+				return map[int64][]*querypb.FieldIndexInfo{2: {
 					{
 						FieldID:     101,
 						IndexID:     1000,
 						EnableIndex: false,
 					},
-				}
+				}}, nil
 			}
-			if segmentID == 3 {
-				return []*querypb.FieldIndexInfo{
+			if segmentIDs[0] == 3 {
+				return map[int64][]*querypb.FieldIndexInfo{3: {
 					{
 						FieldID:     101,
 						IndexID:     1002,
 						EnableIndex: false,
 					},
-				}
+				}}, nil
 			}
-			return nil
-		}, nil)
+			return nil, nil
+		})
 
 	suite.broker.EXPECT().ListIndexes(mock.Anything, int64(1)).Return([]*indexpb.IndexInfo{
 		{
@@ -298,23 +298,21 @@ func (suite *IndexCheckerSuite) TestCreateNewIndex() {
 			}, nil
 		},
 	)
-	suite.broker.EXPECT().GetIndexInfo(mock.Anything, mock.Anything, mock.AnythingOfType("int64")).Call.
-		Return(func(ctx context.Context, collectionID, segmentID int64) []*querypb.FieldIndexInfo {
-			return []*querypb.FieldIndexInfo{
-				{
-					FieldID:        101,
-					IndexID:        1000,
-					EnableIndex:    true,
-					IndexFilePaths: []string{"index"},
-				},
-				{
-					FieldID:        102,
-					IndexID:        1001,
-					EnableIndex:    true,
-					IndexFilePaths: []string{"index"},
-				},
-			}
-		}, nil)
+	suite.broker.EXPECT().GetIndexInfo(mock.Anything, mock.Anything, mock.AnythingOfType("int64")).
+		Return(map[int64][]*querypb.FieldIndexInfo{2: {
+			{
+				FieldID:        101,
+				IndexID:        1000,
+				EnableIndex:    true,
+				IndexFilePaths: []string{"index"},
+			},
+			{
+				FieldID:        102,
+				IndexID:        1001,
+				EnableIndex:    true,
+				IndexFilePaths: []string{"index"},
+			},
+		}}, nil)
 
 	tasks := checker.Check(context.Background())
 	suite.Len(tasks, 1)
diff --git a/internal/querycoordv2/meta/coordinator_broker.go b/internal/querycoordv2/meta/coordinator_broker.go
index f6e6d3fba0..e6ab9a104f 100644
--- a/internal/querycoordv2/meta/coordinator_broker.go
+++ b/internal/querycoordv2/meta/coordinator_broker.go
@@ -48,7 +48,7 @@ type Broker interface {
 	GetRecoveryInfo(ctx context.Context, collectionID UniqueID, partitionID UniqueID) ([]*datapb.VchannelInfo, []*datapb.SegmentBinlogs, error)
 	ListIndexes(ctx context.Context, collectionID UniqueID) ([]*indexpb.IndexInfo, error)
 	GetSegmentInfo(ctx context.Context, segmentID ...UniqueID) ([]*datapb.SegmentInfo, error)
-	GetIndexInfo(ctx context.Context, collectionID UniqueID, segmentID UniqueID) ([]*querypb.FieldIndexInfo, error)
+	GetIndexInfo(ctx context.Context, collectionID UniqueID, segmentIDs ...UniqueID) (map[int64][]*querypb.FieldIndexInfo, error)
 	GetRecoveryInfoV2(ctx context.Context, collectionID UniqueID, partitionIDs ...UniqueID) ([]*datapb.VchannelInfo, []*datapb.SegmentInfo, error)
 	DescribeDatabase(ctx context.Context, dbName string) (*rootcoordpb.DescribeDatabaseResponse, error)
 	GetCollectionLoadInfo(ctx context.Context, collectionID UniqueID) ([]string, int64, error)
@@ -306,13 +306,13 @@ func (broker *CoordinatorBroker) GetSegmentInfo(ctx context.Context, ids ...Uniq
 	return ret, nil
 }
 
-func (broker *CoordinatorBroker) GetIndexInfo(ctx context.Context, collectionID UniqueID, segmentID UniqueID) ([]*querypb.FieldIndexInfo, error) {
+func (broker *CoordinatorBroker) GetIndexInfo(ctx context.Context, collectionID UniqueID, segmentIDs ...UniqueID) (map[int64][]*querypb.FieldIndexInfo, error) {
 	ctx, cancel := context.WithTimeout(ctx, paramtable.Get().QueryCoordCfg.BrokerTimeout.GetAsDuration(time.Millisecond))
 	defer cancel()
 
 	log := log.Ctx(ctx).With(
 		zap.Int64("collectionID", collectionID),
-		zap.Int64("segmentID", segmentID),
+		zap.Int64s("segmentIDs", segmentIDs),
 	)
 
 	// during rolling upgrade, query coord may connect to datacoord with version 2.2, which will return merr.ErrServiceUnimplemented
@@ -322,7 +322,7 @@ func (broker *CoordinatorBroker) GetIndexInfo(ctx context.Context, collectionID
 	retry.Do(ctx, func() error {
 		resp, err = broker.dataCoord.GetIndexInfos(ctx, &indexpb.GetIndexInfoRequest{
 			CollectionID: collectionID,
-			SegmentIDs:   []int64{segmentID},
+			SegmentIDs:   segmentIDs,
 		})
 
 		if errors.Is(err, merr.ErrServiceUnimplemented) {
@@ -337,32 +337,30 @@ func (broker *CoordinatorBroker) GetIndexInfo(ctx context.Context, collectionID
 	}
 
 	if resp.GetSegmentInfo() == nil {
-		err = merr.WrapErrIndexNotFoundForSegment(segmentID)
-		log.Warn("failed to get segment index info",
+		err = merr.WrapErrIndexNotFoundForSegments(segmentIDs)
+		log.Warn("failed to get segments index info",
 			zap.Error(err))
 		return nil, err
 	}
 
-	segmentInfo, ok := resp.GetSegmentInfo()[segmentID]
-	if !ok || len(segmentInfo.GetIndexInfos()) == 0 {
-		return nil, merr.WrapErrIndexNotFoundForSegment(segmentID)
-	}
-
-	indexes := make([]*querypb.FieldIndexInfo, 0)
-	for _, info := range segmentInfo.GetIndexInfos() {
-		indexes = append(indexes, &querypb.FieldIndexInfo{
-			FieldID:             info.GetFieldID(),
-			EnableIndex:         true, // deprecated, but keep it for compatibility
-			IndexName:           info.GetIndexName(),
-			IndexID:             info.GetIndexID(),
-			BuildID:             info.GetBuildID(),
-			IndexParams:         info.GetIndexParams(),
-			IndexFilePaths:      info.GetIndexFilePaths(),
-			IndexSize:           int64(info.GetSerializedSize()),
-			IndexVersion:        info.GetIndexVersion(),
-			NumRows:             info.GetNumRows(),
-			CurrentIndexVersion: info.GetCurrentIndexVersion(),
-		})
+	indexes := make(map[int64][]*querypb.FieldIndexInfo, 0)
+	for segmentID, segmentInfo := range resp.GetSegmentInfo() {
+		indexes[segmentID] = make([]*querypb.FieldIndexInfo, 0)
+		for _, info := range segmentInfo.GetIndexInfos() {
+			indexes[segmentID] = append(indexes[segmentID], &querypb.FieldIndexInfo{
+				FieldID:             info.GetFieldID(),
+				EnableIndex:         true, // deprecated, but keep it for compatibility
+				IndexName:           info.GetIndexName(),
+				IndexID:             info.GetIndexID(),
+				BuildID:             info.GetBuildID(),
+				IndexParams:         info.GetIndexParams(),
+				IndexFilePaths:      info.GetIndexFilePaths(),
+				IndexSize:           int64(info.GetSerializedSize()),
+				IndexVersion:        info.GetIndexVersion(),
+				NumRows:             info.GetNumRows(),
+				CurrentIndexVersion: info.GetCurrentIndexVersion(),
+			})
+		}
 	}
 
 	return indexes, nil
diff --git a/internal/querycoordv2/meta/coordinator_broker_test.go b/internal/querycoordv2/meta/coordinator_broker_test.go
index 728b430cc6..578b4f3713 100644
--- a/internal/querycoordv2/meta/coordinator_broker_test.go
+++ b/internal/querycoordv2/meta/coordinator_broker_test.go
@@ -443,7 +443,7 @@ func (s *CoordinatorBrokerDataCoordSuite) TestGetIndexInfo() {
 
 		infos, err := s.broker.GetIndexInfo(ctx, collectionID, segmentID)
 		s.NoError(err)
-		s.ElementsMatch(indexIDs, lo.Map(infos, func(info *querypb.FieldIndexInfo, _ int) int64 {
+		s.ElementsMatch(indexIDs, lo.Map(infos[segmentID], func(info *querypb.FieldIndexInfo, _ int) int64 {
 			return info.GetIndexID()
 		}))
 		s.resetMock()
diff --git a/internal/querycoordv2/meta/mock_broker.go b/internal/querycoordv2/meta/mock_broker.go
index 40618e2ace..23b1b15f28 100644
--- a/internal/querycoordv2/meta/mock_broker.go
+++ b/internal/querycoordv2/meta/mock_broker.go
@@ -214,29 +214,36 @@ func (_c *MockBroker_GetCollectionLoadInfo_Call) RunAndReturn(run func(context.C
 	return _c
 }
 
-// GetIndexInfo provides a mock function with given fields: ctx, collectionID, segmentID
-func (_m *MockBroker) GetIndexInfo(ctx context.Context, collectionID int64, segmentID int64) ([]*querypb.FieldIndexInfo, error) {
-	ret := _m.Called(ctx, collectionID, segmentID)
+// GetIndexInfo provides a mock function with given fields: ctx, collectionID, segmentIDs
+func (_m *MockBroker) GetIndexInfo(ctx context.Context, collectionID int64, segmentIDs ...int64) (map[int64][]*querypb.FieldIndexInfo, error) {
+	_va := make([]interface{}, len(segmentIDs))
+	for _i := range segmentIDs {
+		_va[_i] = segmentIDs[_i]
+	}
+	var _ca []interface{}
+	_ca = append(_ca, ctx, collectionID)
+	_ca = append(_ca, _va...)
+	ret := _m.Called(_ca...)
 
 	if len(ret) == 0 {
 		panic("no return value specified for GetIndexInfo")
 	}
 
-	var r0 []*querypb.FieldIndexInfo
+	var r0 map[int64][]*querypb.FieldIndexInfo
 	var r1 error
-	if rf, ok := ret.Get(0).(func(context.Context, int64, int64) ([]*querypb.FieldIndexInfo, error)); ok {
-		return rf(ctx, collectionID, segmentID)
+	if rf, ok := ret.Get(0).(func(context.Context, int64, ...int64) (map[int64][]*querypb.FieldIndexInfo, error)); ok {
+		return rf(ctx, collectionID, segmentIDs...)
 	}
-	if rf, ok := ret.Get(0).(func(context.Context, int64, int64) []*querypb.FieldIndexInfo); ok {
-		r0 = rf(ctx, collectionID, segmentID)
+	if rf, ok := ret.Get(0).(func(context.Context, int64, ...int64) map[int64][]*querypb.FieldIndexInfo); ok {
+		r0 = rf(ctx, collectionID, segmentIDs...)
 	} else {
 		if ret.Get(0) != nil {
-			r0 = ret.Get(0).([]*querypb.FieldIndexInfo)
+			r0 = ret.Get(0).(map[int64][]*querypb.FieldIndexInfo)
 		}
 	}
 
-	if rf, ok := ret.Get(1).(func(context.Context, int64, int64) error); ok {
-		r1 = rf(ctx, collectionID, segmentID)
+	if rf, ok := ret.Get(1).(func(context.Context, int64, ...int64) error); ok {
+		r1 = rf(ctx, collectionID, segmentIDs...)
 	} else {
 		r1 = ret.Error(1)
 	}
@@ -252,24 +259,31 @@ type MockBroker_GetIndexInfo_Call struct {
 // GetIndexInfo is a helper method to define mock.On call
 //   - ctx context.Context
 //   - collectionID int64
-//   - segmentID int64
-func (_e *MockBroker_Expecter) GetIndexInfo(ctx interface{}, collectionID interface{}, segmentID interface{}) *MockBroker_GetIndexInfo_Call {
-	return &MockBroker_GetIndexInfo_Call{Call: _e.mock.On("GetIndexInfo", ctx, collectionID, segmentID)}
+//   - segmentIDs ...int64
+func (_e *MockBroker_Expecter) GetIndexInfo(ctx interface{}, collectionID interface{}, segmentIDs ...interface{}) *MockBroker_GetIndexInfo_Call {
+	return &MockBroker_GetIndexInfo_Call{Call: _e.mock.On("GetIndexInfo",
+		append([]interface{}{ctx, collectionID}, segmentIDs...)...)}
 }
 
-func (_c *MockBroker_GetIndexInfo_Call) Run(run func(ctx context.Context, collectionID int64, segmentID int64)) *MockBroker_GetIndexInfo_Call {
+func (_c *MockBroker_GetIndexInfo_Call) Run(run func(ctx context.Context, collectionID int64, segmentIDs ...int64)) *MockBroker_GetIndexInfo_Call {
 	_c.Call.Run(func(args mock.Arguments) {
-		run(args[0].(context.Context), args[1].(int64), args[2].(int64))
+		variadicArgs := make([]int64, len(args)-2)
+		for i, a := range args[2:] {
+			if a != nil {
+				variadicArgs[i] = a.(int64)
+			}
+		}
+		run(args[0].(context.Context), args[1].(int64), variadicArgs...)
 	})
 	return _c
 }
 
-func (_c *MockBroker_GetIndexInfo_Call) Return(_a0 []*querypb.FieldIndexInfo, _a1 error) *MockBroker_GetIndexInfo_Call {
+func (_c *MockBroker_GetIndexInfo_Call) Return(_a0 map[int64][]*querypb.FieldIndexInfo, _a1 error) *MockBroker_GetIndexInfo_Call {
 	_c.Call.Return(_a0, _a1)
 	return _c
 }
 
-func (_c *MockBroker_GetIndexInfo_Call) RunAndReturn(run func(context.Context, int64, int64) ([]*querypb.FieldIndexInfo, error)) *MockBroker_GetIndexInfo_Call {
+func (_c *MockBroker_GetIndexInfo_Call) RunAndReturn(run func(context.Context, int64, ...int64) (map[int64][]*querypb.FieldIndexInfo, error)) *MockBroker_GetIndexInfo_Call {
 	_c.Call.Return(run)
 	return _c
 }
diff --git a/internal/querycoordv2/task/executor.go b/internal/querycoordv2/task/executor.go
index 764a7afd4f..36f04889d2 100644
--- a/internal/querycoordv2/task/executor.go
+++ b/internal/querycoordv2/task/executor.go
@@ -703,7 +703,7 @@ func (ex *Executor) getLoadInfo(ctx context.Context, collectionID, segmentID int
 		return nil, nil, err
 	}
 	// update the field index params
-	for _, segmentIndex := range indexes {
+	for _, segmentIndex := range indexes[segment.GetID()] {
 		index, found := lo.Find(indexInfos, func(indexInfo *indexpb.IndexInfo) bool {
 			return indexInfo.IndexID == segmentIndex.IndexID
 		})
@@ -720,6 +720,6 @@ func (ex *Executor) getLoadInfo(ctx context.Context, collectionID, segmentID int
 		segmentIndex.IndexParams = funcutil.Map2KeyValuePair(params)
 	}
 
-	loadInfo := utils.PackSegmentLoadInfo(segment, channel.GetSeekPosition(), indexes)
+	loadInfo := utils.PackSegmentLoadInfo(segment, channel.GetSeekPosition(), indexes[segment.GetID()])
 	return loadInfo, indexInfos, nil
 }
diff --git a/internal/querycoordv2/task/task_test.go b/internal/querycoordv2/task/task_test.go
index e1d3e50c9f..46a775e1ac 100644
--- a/internal/querycoordv2/task/task_test.go
+++ b/internal/querycoordv2/task/task_test.go
@@ -532,7 +532,7 @@ func (suite *TaskSuite) TestLoadSegmentTaskNotIndex() {
 				InsertChannel: channel.ChannelName,
 			},
 		}, nil)
-		suite.broker.EXPECT().GetIndexInfo(mock.Anything, suite.collection, segment).Return(nil, merr.WrapErrIndexNotFoundForSegment(segment))
+		suite.broker.EXPECT().GetIndexInfo(mock.Anything, suite.collection, segment).Return(nil, merr.WrapErrIndexNotFoundForSegments([]int64{segment}))
 	}
 	suite.cluster.EXPECT().LoadSegments(mock.Anything, targetNode, mock.Anything).Return(merr.Success(), nil)
 
diff --git a/pkg/util/merr/errors_test.go b/pkg/util/merr/errors_test.go
index 125a2e72f9..4febe3e64c 100644
--- a/pkg/util/merr/errors_test.go
+++ b/pkg/util/merr/errors_test.go
@@ -120,7 +120,7 @@ func (s *ErrSuite) TestWrap() {
 	// Index related
 	s.ErrorIs(WrapErrIndexNotFound("failed to get Index"), ErrIndexNotFound)
 	s.ErrorIs(WrapErrIndexNotFoundForCollection("milvus_hello", "failed to get collection index"), ErrIndexNotFound)
-	s.ErrorIs(WrapErrIndexNotFoundForSegment(100, "failed to get collection index"), ErrIndexNotFound)
+	s.ErrorIs(WrapErrIndexNotFoundForSegments([]int64{100}, "failed to get collection index"), ErrIndexNotFound)
 	s.ErrorIs(WrapErrIndexNotSupported("wsnh", "failed to create index"), ErrIndexNotSupported)
 
 	// Node related
diff --git a/pkg/util/merr/utils.go b/pkg/util/merr/utils.go
index 2ad30564e5..3ded3f7029 100644
--- a/pkg/util/merr/utils.go
+++ b/pkg/util/merr/utils.go
@@ -777,8 +777,8 @@ func WrapErrIndexNotFound(indexName string, msg ...string) error {
 	return err
 }
 
-func WrapErrIndexNotFoundForSegment(segmentID int64, msg ...string) error {
-	err := wrapFields(ErrIndexNotFound, value("segmentID", segmentID))
+func WrapErrIndexNotFoundForSegments(segmentIDs []int64, msg ...string) error {
+	err := wrapFields(ErrIndexNotFound, value("segmentIDs", segmentIDs))
 	if len(msg) > 0 {
 		err = errors.Wrap(err, strings.Join(msg, "->"))
 	}