Fixed issue with model.