Fix build
Merge pull request #19 from YouweGit/fix-build

Fix build