Fix bug
Merge pull request #252 from dreamsxin/model

Fix bug