update tests to the new ui changes.