fixed some of the model tests