Fix build
Merge pull request #5 from sarnex/master

Fix build