Added missing dependancies in controller