Fixing integration test