Corrected unit test build error