fix UT assertion