Fix UI Test error