Change class to controller in service