Fix Deprecation Warning

This simply fixes a deprecation warning in the upcoming 0.44 release. :)