fix the tests.....