fixed some of the models