Fix ошибок

Выловил несколько ошибок в коде из-за рефакторинга модели