fix UT build error