Fix a failing integration test