Fix in the tests