Fixing integration tests.