Fix #30
Merge pull request #33 from yongkangchen/patch-2

Fix #30