Fixed one more problem with the model.