Fix issue with controller file