Minor fix on python impl