Fix build
Merge pull request #11 from nlegoff/fix_build

Fix build