Fix unit test compilation failure