fix the undefined error in UT