Fix syntax for python2