Fix the issue in ut.