UT bug fix