fix UT issue