Fixed a bit bug in prev commit
Merge pull request #32 from kouyaf77/master

Fixed a bit bug in prev commit