Fix bug with Python3