UT refactoring