fix for the model