[E2E] Fix 13455 flake (#21476)

* Move comment to an appropriate place

* Limit the number of results to speed things up

* Wait for both `dataset` responses to come back

* Use second to last index

* Use a helper function instead of aliases

* Add some additional comments

* Update test title