Fix UI test