Fix failing build due to upstream changes.
Merge pull request #38 from agouge/fix/upstream-changes

Quick Fix (one line)