Update to Response object for testing