Small cosmetic edit