Actually fix the deprecation error this time :)