update tests to reflect new model api