fixed the tests