Fix: update to new API
Merge pull request #1 from gram003/master

Fix: update to new API