Fix the integration tests.