Fix compile error
Merge pull request #8 from tibel/master

Fix compile error