Fix compilation
Merge pull request #2 from tindzk/master

Fix compilation