Fixed a bug in the models