fix unit tests.