Fix integration test.