Fix the bug in UT.