Fix Python 3 incompatability