Switch to the class-based approach for controllers