Correct models to according to database