Made changes to one of the controller tests