diff --git a/internal/dataservice/grpc_services.go b/internal/dataservice/grpc_services.go
index b296df6c81..b125f185f8 100644
--- a/internal/dataservice/grpc_services.go
+++ b/internal/dataservice/grpc_services.go
@@ -92,6 +92,8 @@ func (s *Server) AssignSegmentID(ctx context.Context, req *datapb.AssignSegmentI
 		//continue
 		//}
 
+		s.cluster.watchIfNeeded(r.ChannelName, r.CollectionID)
+
 		segmentID, retCount, expireTs, err := s.segAllocator.AllocSegment(ctx,
 			r.CollectionID, r.PartitionID, r.ChannelName, int64(r.Count))
 		if err != nil {
diff --git a/internal/dataservice/server.go b/internal/dataservice/server.go
index 7899ec212a..42cf67580c 100644
--- a/internal/dataservice/server.go
+++ b/internal/dataservice/server.go
@@ -542,7 +542,8 @@ func (s *Server) loadCollectionFromMaster(ctx context.Context, collectionID int6
 		CollectionID:   resp.CollectionID,
 	})
 	if err = VerifyResponse(presp, err); err != nil {
-		log.Error("show partitions error", zap.String("collectionName", resp.Schema.Name), zap.Int64("collectionID", resp.CollectionID), zap.Error(err))
+		log.Error("show partitions error", zap.String("collectionName", resp.Schema.Name),
+			zap.Int64("collectionID", resp.CollectionID), zap.Error(err))
 		return err
 	}
 	collInfo := &datapb.CollectionInfo{