Fix iterater bug