From 0ff79dc4e42876f4bb092db48fd4552a9c27326e Mon Sep 17 00:00:00 2001
From: Hyphie <31455641+Hyphie24@users.noreply.github.com>
Date: Fri, 18 Jul 2025 10:42:52 -0400
Subject: [PATCH] Add AI Class colors to UnitFrame

---
 modules/health.lua | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/health.lua b/modules/health.lua
index f9081cf5..48c7ceeb 100755
--- a/modules/health.lua
+++ b/modules/health.lua
@@ -115,7 +115,7 @@ function Health:UpdateColor(frame)
 				color = ShadowUF.db.profile.healthColors.hostile
 			end
 		end
-	elseif( ShadowUF.db.profile.units[frame.unitType].healthBar.colorType == "class" and (UnitIsPlayer(unit) or unit == "pet") ) then
+	elseif( ShadowUF.db.profile.units[frame.unitType].healthBar.colorType == "class" and (UnitIsPlayer(unit) or UnitInPartyIsAI(unit) or unit == "pet") ) then
 		local class = (unit == "pet") and "PET" or frame:UnitClassToken()
 		color = class and ShadowUF.db.profile.classColors[class]
 	elseif( ShadowUF.db.profile.units[frame.unitType].healthBar.colorType == "playerclass" and unit == "pet") then