Improved the integration test.