fixed the unit tests.