#27343 fixed someflatiter indexing issues. However, as mentioned on the PR, there's still cases where indexing aflatiter is incorrect. The correct solution here would be to re-use index preparation code, so that indexing is consistent throughoutflatiters andndarrays.