diff --git a/IceUnitBar.lua b/IceUnitBar.lua index c9d5405..82f8fe4 100644 --- a/IceUnitBar.lua +++ b/IceUnitBar.lua @@ -46,6 +46,7 @@ end function IceUnitBar.prototype:SetUnit(unit) self.unit = unit + local _ _, self.unitClass = UnitClass(self.unit) end diff --git a/modules/TargetInfo.lua b/modules/TargetInfo.lua index fd07389..f36b580 100644 --- a/modules/TargetInfo.lua +++ b/modules/TargetInfo.lua @@ -107,6 +107,7 @@ end function IceTargetInfo.prototype:Enable(core) IceTargetInfo.super.prototype.Enable(self, core) + local _ _, self.playerClass = UnitClass("player") if IceHUD.IceCore:ShouldUseDogTags() then