changed the model to the correct one