Merge pull request #7 from peterlee0127/master

fix