Fix compile error
Merge pull request #4 from sakana/master

Fix compile error