fix thanks to Jake
Merge pull request #9 from highpowerxh/master

A fix thanks to Jake