UI test fix