diff --git a/internal/querynodev2/metrics_info.go b/internal/querynodev2/metrics_info.go
index f0429d8960..248d0b136e 100644
--- a/internal/querynodev2/metrics_info.go
+++ b/internal/querynodev2/metrics_info.go
@@ -116,10 +116,8 @@ func getQuotaMetrics(node *QueryNode) (*metricsinfo.QueryNodeQuotaMetrics, error
 		metrics.QueryNodeEntitiesSize.WithLabelValues(fmt.Sprint(node.GetNodeID()),
 			fmt.Sprint(collection), segments.SegmentTypeSealed.String()).Set(float64(size))
 	}
-	sealedGroupByPartition := lo.GroupBy(sealedSegments, func(seg segments.Segment) int64 {
-		return seg.Partition()
-	})
-	for _, segs := range sealedGroupByPartition {
+
+	for _, segs := range sealedGroupByCollection {
 		numEntities := lo.SumBy(segs, func(seg segments.Segment) int64 {
 			return seg.RowNum()
 		})