diff --git a/elements/totems.lua b/elements/totems.lua index b8541414..eb0ecb89 100644 --- a/elements/totems.lua +++ b/elements/totems.lua @@ -48,6 +48,11 @@ OnEnter and OnLeave script handlers will be set to display a Tooltip if the `Tot local _, ns = ... local oUF = ns.oUF +local TOTEM_PRIORITIES = _G.STANDARD_TOTEM_PRIORITIES +if(UnitClassBase('player') == 'SHAMAN') then + TOTEM_PRIORITIES = _G.SHAMAN_TOTEM_PRIORITIES +end + local function UpdateTooltip(self) GameTooltip:SetTotem(self:GetID()) end @@ -75,7 +80,7 @@ local function UpdateTotem(self, event, slot) --]] if(element.PreUpdate) then element:PreUpdate(slot) end - local totem = element[slot] + local totem = element[TOTEM_PRIORITIES[slot]] local haveTotem, name, start, duration, icon = GetTotemInfo(slot) if(haveTotem and duration > 0) then if(totem.Icon) then @@ -135,8 +140,7 @@ local function Enable(self) element.ForceUpdate = ForceUpdate for i = 1, #element do - local totem = element[i] - + local totem = element[TOTEM_PRIORITIES[i]] totem:SetID(i) if(totem:IsMouseEnabled()) then