Fixed a bug in the model