WoWInterface - Highlight dispellable buffs (target/focus) for non-purge classes

Hi All,

I've been using this bit of code all of DF to highlight dispellable buffs on the target/focus frame for classes without a dispel (found here from SDPhantom):

Lua Code:

local function TargetFrame_UpdateAuras(self)
    for buff in self.auraPools:GetPool("TargetBuffFrameTemplate"):EnumerateActive() do
        local data=C_UnitAuras.GetAuraDataByAuraInstanceID(buff.unit,buff.auraInstanceID);
        buff.Stealable:SetShown(data.isStealable or data.dispelName=="Magic");
    end
end
 
hooksecurefunc(TargetFrame,"UpdateAuras",TargetFrame_UpdateAuras);
hooksecurefunc(FocusFrame,"UpdateAuras",TargetFrame_UpdateAuras);

However, since 10.2.5, I've been using Jax Classic Frames (https://github.com/skidsh/JaxClassicFrames) and the code above no longer works.

Would anyone know how I could edit the above code or the code within Jax Classic Frames to highlight dispellable buffs on the target/focus frame for all classes (rather than only classes with a dispel)?

Any help would greatly be appreciated!

Thank you for your time :)