Changing structure of ControllerTest class. Not working.