Fix UT after the v1.0.1