Fix failing ui test.