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

Fix compilation