Fix Python 2 support.