fixed issue #4
Merge pull request #21 from leonxlin/master

fixed issue #4