Поправлен контроллер