Yes but I was implying do this for Python2 as well then you don't depend on the deprecated function and we can use it =)