Does anyone know how to change portrait to SQUARE Class Icons? Particularly for the ArenaEnemyFrames1-3.

Code:
/run UFP = "UnitFramePortrait_Update"; UICC = "Interface\\Glues\\CharacterCreate\\UI-CharacterCreate-Classes"; CIT = CLASS_ICON_TCOORDS

/run hooksecurefunc(UFP,function(self) if self.portrait then local t = CIT[select(2,UnitClass(self.unit))] if t then self.portrait:SetTexture(UICC) self.portrait:SetTexCoord(unpack(t)) end end end)
This changes it to square icons but it doesn't seem to work for AEFs. Also pets and totems seem to be wrong.

Anyone know how to do this?