Fix incorrect UT