Fix typing error in Model update