Fixing unit test build.