Fix unit test compile error.