Fix python bugs