I'm also wondering if we should apply this strategy if any of the segments matches this criterion. This way we wouldn't need the `f (segmentDocCounts != null)` condition in `collect` anymore?