Fixed the unit test build.