fix Issue #2
Merge pull request #4 from xaben/fix-model

fix Issue #2