Fix refactoring custom controller actions