fix the tests