diff --git a/internal/datacoord/segment_info.go b/internal/datacoord/segment_info.go
index d4a25ff745..1ca48d5da0 100644
--- a/internal/datacoord/segment_info.go
+++ b/internal/datacoord/segment_info.go
@@ -78,7 +78,7 @@ func (s *SegmentsInfo) SetRowCount(segmentID UniqueID, rowCount int64) {
 	}
 }
 
-// SetStates sets Segment State info for SegmentInfo with provided segmentID
+// SetState sets Segment State info for SegmentInfo with provided segmentID
 // if SegmentInfo not found, do nothing
 func (s *SegmentsInfo) SetState(segmentID UniqueID, state commonpb.SegmentState) {
 	if segment, ok := s.segments[segmentID]; ok {