Hotfix for deprecation
Merge pull request #76 from zefciu/deprecation-hotfix

Hotfix for deprecation