Fix failing integration test.