Fix UT failed by reverting code committing.