fix the tests.