fix python example