diff --git a/internal/datacoord/segment_info.go b/internal/datacoord/segment_info.go
index 0e0bebb1b1..cd2d0a045e 100644
--- a/internal/datacoord/segment_info.go
+++ b/internal/datacoord/segment_info.go
@@ -224,6 +224,7 @@ func SetStartPosition(pos *internalpb.MsgPosition) SegmentInfoOption {
 	}
 }
 
+// SetAllocations is the option to set allocations for segment info
 func SetAllocations(allocations []*Allocation) SegmentInfoOption {
 	return func(segment *SegmentInfo) {
 		segment.allocations = allocations