Fix iterator mis-use