Small iterator fix