Maybe this will fix the DeprecationWarning