Update tests to match latest UI changes.
2 files changed