chore(tests): follow new model definition