update the tests following the changes