Fix/model bug