add upstream fix
Merge pull request #3 from leigh123linux/master

add upstream fix