Fix reference to data in controller