Fix failed UT