fix #3
Merge pull request #4 from GutenYe/fix-model

fix #3