small fix on the tests