fix UT bug