Fix reference to controller