fix bug
Merge pull request #62 from PuZheng/issueX

fix bug