UI test fixes