Small cosmetic UI fix