Fix a little bug
Merge pull request #4 from djhvscf/master

Fix a little bug