fix build error of pull request #5
Merge pull request #6 from calorie/fix_build_error_of_#5

fix build error of pull request #5