diff --git a/internal/querynodev2/metrics_info.go b/internal/querynodev2/metrics_info.go
index 57f98606a6..36514a9967 100644
--- a/internal/querynodev2/metrics_info.go
+++ b/internal/querynodev2/metrics_info.go
@@ -92,11 +92,8 @@ func getQuotaMetrics(node *QueryNode) (*metricsinfo.QueryNodeQuotaMetrics, error
 		metrics.QueryNodeEntitiesSize.WithLabelValues(nodeID, fmt.Sprint(collection),
 			segments.SegmentTypeGrowing.String()).Set(float64(size))
 	}
-	growingGroupByPartition := lo.GroupBy(growingSegments, func(seg segments.Segment) int64 {
-		return seg.Partition()
-	})
 
-	for _, segs := range growingGroupByPartition {
+	for _, segs := range growingGroupByCollection {
 		numEntities := lo.SumBy(segs, func(seg segments.Segment) int64 {
 			return seg.RowNum()
 		})