Fix the UI test.