fix up the model