Fix a bug on the tests