fix a bug in #77
Merge pull request #78 from bachue/fix-bug-for-77

fix a bug in #77