Fix deprecation warning
Merge pull request #2 from jhnvz/fix/deprecation-warning

Fix deprecation warning