Fix compiler error in UT