Update test_ui.py