fixed a bug in models
Merge pull request #5 from WesGBrooks/develop

fixed a bug in models