diff --git a/HealBot_Controller_Spells.lua b/HealBot_Controller_Spells.lua
index ae95946..6cfa3f6 100644
--- a/HealBot_Controller_Spells.lua
+++ b/HealBot_Controller_Spells.lua
@@ -510,7 +510,7 @@ function HealBot_FindHealSpells()
           temp_Spell_cast = HealBot_Spells[spell].CastTime;
         end
         RealHealing = ((HealBot_GetBonus() * healingbonus_penalty) * (temp_Spell_cast / 3.5));
-        local playerClass, englishClass = UnitClass("player");
+        local englishClass = HealBot_UnitClass("player");
         local SpiBonus = 0;
         if (englishClass == "PRIEST") then
           SpiBonus = ((HealBot_SpiBonus(spell) * healingbonus_penalty) * (temp_Spell_cast / 3.5))
diff --git a/HealBot_Model.lua b/HealBot_Model.lua
index 6ea30e0..7ebfc74 100644
--- a/HealBot_Model.lua
+++ b/HealBot_Model.lua
@@ -149,7 +149,7 @@ function HealBot_Model:UpdateUnitIdentity(unit)
     local oldName = self.units[unit].name
     local oldEnglishClass = self.units[unit].englishClass
     local name = UnitName(unit)
-    local _, englishClass = UnitClass(unit)
+    local englishClass = HealBot_UnitClass(unit)
     
     if oldName ~= name or oldEnglishClass ~= englishClass then
         self.units[unit].name = name