Fix the issue in Ut.