Fixed the pointed out
Merge pull request #6 from xymz/pr_comment

Fixed the pointed out