Minor Aesthetic fix