Fix for Python 2.