Fixed bug in new model.