[No reviewer] Fix incorrect UT