Minor edit to correct python2ism