Fix unit test compile error