fix iterator prob