Update for the new get_controller.