Update test to reflect changes to UI.