update for API change
Merge pull request #124 from teharrison/master

update for API change