Fix Deprecation Warning in upcoming 0.44 release