diff --git a/mods/unitframes-classcolor.lua b/mods/unitframes-classcolor.lua
index 5993d53..acc2e4d 100644
--- a/mods/unitframes-classcolor.lua
+++ b/mods/unitframes-classcolor.lua
@@ -25,7 +25,7 @@ module.enable = function(self)
 
     local reaction = UnitReaction("target", "player")
 
-	  if UnitPlayerControlled("target") then
+	  if UnitIsPlayer("target") then
 	    local _, class = UnitClass("target")
 	    local class = RAID_CLASS_COLORS[class] or { r = .5, g = .5, b = .5, a = 1 }
 	    TargetFrameNameBackground:SetVertexColor(class.r, class.g, class.b, 1)