Fix UT errors