Aesthetic bug fix