fix UT case