- fixed the tests