Fix test bug where multi-segment indexes weren't properly collected