fix the isuue in UT.