fix compile
Merge pull request #7 from safetytrick/master

fix compile