Update ui_test.py