fixing the issue #4 (DeprecationWarning)