changed to include the model fix