fix bug 

fix bug
Merge pull request #67 from wenjun1055/patch-1

fix bug