Cosmetic update

Minor cosmetic update