Fixed the previous commit for Python3.