modify the unit tests.