diff --git a/internal/datacoord/segment_info.go b/internal/datacoord/segment_info.go
index 18027244fb..8530b64f97 100644
--- a/internal/datacoord/segment_info.go
+++ b/internal/datacoord/segment_info.go
@@ -186,6 +186,7 @@ func (s *SegmentInfo) ShadowClone(opts ...SegmentInfoOption) *SegmentInfo {
 // SegmentInfoOption is the option to set fields in segment info
 type SegmentInfoOption func(segment *SegmentInfo)
 
+// SetRowCount is the option to set row count for segment info
 func SetRowCount(rowCount int64) SegmentInfoOption {
 	return func(segment *SegmentInfo) {
 		segment.NumOfRows = rowCount