Small python change.