Fixed the unit test build