Minor fix
Minor fixes