Fix integration tests.