Fix compilation.
Merge pull request #90 from Arkshine/fix-compilation

Fix compilation.