Fix #20 with the recent API change
Merge pull request #39 from matthew-andrews/20

Fix #20 with the recent API change