fix build failure
Merge pull request #10 from josephyzhou/master

fix build failure