Fix the compile error
Merge pull request #16 from jaccarmac/fix-error

Fix the compile error