Invalid iterator fix