small change on the python wrapper