fix UT error