Немного дополнил контролер для тестов