little update in the tests