Fix Python 3.x compatibility