Fix python example [skip ci]