fix the tests
fix the tests