update tests to reflect change in custom methods