Fix cosmetic bug