Решал через итератор, похоже чего- то не учел