unit test build fix