fixing Test to match new UI.