Fix compile error in ut