fixed the issue
Merge pull request #14 from katexyu/master

Fixed Issue #4