fixes the tests