Minor fix for setup py