Fix model issu