simplify loop

Matches the new loop below it