diff --git a/src/services/wiki_masters.py b/src/services/wiki_masters.py index a704ada..aac0d5b 100644 --- a/src/services/wiki_masters.py +++ b/src/services/wiki_masters.py @@ -26,12 +26,12 @@ class Rarity(StrEnum): @property def icon(self) -> str: return { - Self.Common: "⚪", - Self.LessCommon: "🟢", - Self.Rare: "🔵", - Self.SuperRare: "🟣", - Self.UltraRare: "🟠", - Self.Legendary: "🌟", + Rarity.Common: "⚪", + Rarity.LessCommon: "🟢", + Rarity.Rare: "🔵", + Rarity.SuperRare: "🟣", + Rarity.UltraRare: "🟠", + Rarity.Legendary: "🌟", }[self] @dataclass