fix
Merge pull request #8 from wraith228/fix

fix