Fix iteration tests