Minor fixes
Minor fix