TESTS - Update tests to match the new ui.