Fixed unit tests build.
Fixed the unit tests.