fix a UT function