Fix test for the new UI.