fix the UT issue.