Fix a problem with the tests