Minor cosmetic update
Minor fix