Fix failing integration test