Fix typo for python2