diff --git a/modules/buff.lua b/modules/buff.lua
index cf5a8092..10ee345b 100644
--- a/modules/buff.lua
+++ b/modules/buff.lua
@@ -138,16 +138,10 @@ pfUI:RegisterModule("buff", function ()
       end
     end)
 
-    buff:SetScript("OnLeave", function()
-      GameTooltip:Hide()
-    end)
+    buff:SetScript("OnLeave", GameTooltip_Hide)
 
     buff:SetScript("OnClick", function()
-      if CancelItemTempEnchantment and this.mode and this.mode == "MAINHAND" then
-        CancelItemTempEnchantment(1)
-      elseif CancelItemTempEnchantment and this.mode and this.mode == "OFFHAND" then
-        CancelItemTempEnchantment(2)
-      elseif this.spellId then
+      if this.spellId then
         C_Spell.CancelSpellByID(this.spellId)
       end
     end)