small update to model.py [MJ-677]