Changing structure of ControllerTest class.