small fix for the tests