add to model tests