Fix the issue in UT.