Fix unit test build