Fix compile error in unit test