Make small refactoring in controllers.