Correct the model which is being imported