Исправление обработки неправильного класса.