Fix bug with new model.