Small change to controller.