updating unit tests
fixing the tests