Added missing modification to controller.