diff --git a/internal/datacoord/segment_info.go b/internal/datacoord/segment_info.go
index 8530b64f97..e290198dac 100644
--- a/internal/datacoord/segment_info.go
+++ b/internal/datacoord/segment_info.go
@@ -200,6 +200,7 @@ func SetExpireTime(expireTs Timestamp) SegmentInfoOption {
 	}
 }
 
+// SetState is the option to set state for segment info
 func SetState(state commonpb.SegmentState) SegmentInfoOption {
 	return func(segment *SegmentInfo) {
 		segment.State = state