Fix the tests

same here