fixing flat UI test cases