fix python bug