fix the tests following revert