fix bug
Merge pull request #2 from poppinlp/master

fix bug