Fixed the TestController to work with the response objects.