Got the integration test to pass