Fixed the models