fix the issue in ut.