Fix build
Merge pull request #4 from carlyeks/fix-build

Fix build