Fix Deprecation Warning

This simply fixes a deprecation warning in the upcoming 0.44 release. Tested, everything appears to still work (aside from #4). I was not quite sure how you are formatting the code, so let me know if it needs un/indenting. :)