Fix minor bug in Model