Fix build
Merge pull request #2 from swills/master

Fix build