Fix Python 3 issue