Fixed the integration test.