trying to fix iterator