Update "with" hook to be class method, not function (and test).