portrait: Use UnitClass to get accurate info on AI players (#691)

oUF-wow · Nov 30, 2024 · b41046b · b41046b
1 parent 69b0657
commit b41046b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/elements/portrait.lua b/elements/portrait.lua
@@ -71,7 +71,13 @@ local function Update(self, event, unit)
 				element:SetUnit(unit)
 			end
 		else
-			local class = element.showClass and UnitClassBase(unit)
+			local class, _
+			if(element.showClass) then
+				-- BUG: UnitClassBase can't be trusted
+				--      https://github.com/Stanzilla/WoWUIBugs/issues/621
+				_, class = UnitClass(unit)
+			end
+
 			if(class) then
 				element:SetAtlas('classicon-' .. class)
 			else