Исправление ошибки и рефакторинг