Fix unit tests build issue