fix after changed API
Merge pull request #1 from mhamrle/changed_api

fix after changed API