Fix unit test compllation