More UI test coverage