Merge pull request #6 from LinZap/master

add fix