Fix deprecation warning
Merge pull request #95 from tribal-tec/master

Fix deprecation warning