Fix compilation
Merge pull request #2 from s-i-newton/fix-compil

Fix compilation