updating tests to match new UI.