[Python] Minor fix