fixes the tests ..