fix compilation error
Merge pull request #30 from littlewhitetwo/master

fix compilation error