Another python 3 fix