Revert "Fix flaky index sampling test."

This reverts commit af2bed3bd20f447e503425da40034cae45443977.