Поправил интерфейс. Убрал костылик =)