Fix broken build
Merge pull request #5 from trevorbernard/master

Fix broken build