Fixing the unit test build failure.