Fix for issue #2.
Merge pull request #3 from hongster/master

Fix for issue #2.