Fix ControllerGenerator