fix build
Merge pull request #4 from l3iggs/patch-1

fix build