Fix minor bug for Python 2.x