Fixing integration test.