Fixed a python2/python3 issue