Fix iterator bug!