Fix unit test build.