Fixed a model bug