updating unit tests.