Make extrabutton logic consistent with zoneability button

brittyazel · Jan 18, 2019 · e7e9a7e · e7e9a7e
1 parent 3c0c0c7
commit e7e9a7e
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/Objects/EXTRABTN.lua b/Objects/EXTRABTN.lua
@@ -92,13 +92,8 @@ function EXTRABTN:ExtraButton_Update()
 	end
 
 	_, self.spellID = GetActionInfo(self.actionID)
-
 	self.spellName, _, self.spellIcon = GetSpellInfo(self.spellID);
 
-	if not InCombatLockdown() then
-		self:SetAttribute("action", self.actionID)
-	end
-
 	if self.spellID then
 		self:SetButtonTex()
 
@@ -107,6 +102,10 @@ function EXTRABTN:ExtraButton_Update()
 		if (start) then
 			self:SetCooldownTimer(start, duration, enable, self.cdText, modrate, self.cdcolor1, self.cdcolor2, self.cdAlpha)
 		end
+
+		if not InCombatLockdown() then
+			self:SetAttribute("action", self.actionID)
+		end
 	end