Minor fix for python arg parsing