fixed small == bug in model