Fix error on Python 2.