fix UT
bug fix