Fix Python 2.x