Fix build
Merge pull request #2 from kevva/fix

Fix build