fix compile
Merge pull request #6 from jshlee/master

fix compile