modify unit tests.