Fix a missing model update