Correct the models