From cb0c5932a6966516656c5e3ce6c8ab0f3b3496d6 Mon Sep 17 00:00:00 2001
From: Peechey <92683202+Peechey@users.noreply.github.com>
Date: Wed, 22 Apr 2026 22:42:01 -0500
Subject: [PATCH 1/2] add modflag to imbued changes

---
 src/Classes/SkillsTab.lua | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/Classes/SkillsTab.lua b/src/Classes/SkillsTab.lua
index 117f19ed1b..e60c561b2b 100644
--- a/src/Classes/SkillsTab.lua
+++ b/src/Classes/SkillsTab.lua
@@ -238,6 +238,7 @@ local SkillsTabClass = newClass("SkillsTab", "UndoHandler", "ControlHost", "Cont
 				self.displayGroup.imbuedSupport = nil
 			end
 		end
+		self.modFlag = true
 	end, true, true)
 	local function isImbuedEnabled() -- socketedIn must be set and the displayGroup must have an imbued, otherwise disable the imbued dropdown
 		return (self.displayGroup and self.displayGroup.slot and ((self.imbuedSupportBySlot[self.displayGroup.slot] and self.displayGroup.imbuedSupport) or not self.imbuedSupportBySlot[self.displayGroup.slot]))
@@ -254,6 +255,7 @@ local SkillsTabClass = newClass("SkillsTab", "UndoHandler", "ControlHost", "Cont
 		self.displayGroup.imbuedSupport = nil
 		self.imbuedSupportBySlot[self.displayGroup.slot] = nil
 		self.build.buildFlag = true
+		self.modFlag = true
 	end)
 	self.controls.imbuedSupportClear.enabled = function()
 		return isImbuedEnabled()

From 1341b5a91226143fdab421044457c002bb8ee80b Mon Sep 17 00:00:00 2001
From: Peechey <92683202+Peechey@users.noreply.github.com>
Date: Wed, 22 Apr 2026 22:52:51 -0500
Subject: [PATCH 2/2] undostate

---
 src/Classes/SkillsTab.lua | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Classes/SkillsTab.lua b/src/Classes/SkillsTab.lua
index e60c561b2b..3f2f6dd0a2 100644
--- a/src/Classes/SkillsTab.lua
+++ b/src/Classes/SkillsTab.lua
@@ -224,6 +224,7 @@ local SkillsTabClass = newClass("SkillsTab", "UndoHandler", "ControlHost", "Cont
 			return
 		end
 		local updateDisplayGroup = self.displayGroup and targetSlot == self.displayGroup.slot
+		self:AddUndoState()
 		if gemData and (type(gemData) == "string" or gemData.id) then
 			local gem = data.gems[gemData.id or gemData]
 			self.imbuedSupportBySlot[targetSlot] = gem.grantedEffect
@@ -238,7 +239,6 @@ local SkillsTabClass = newClass("SkillsTab", "UndoHandler", "ControlHost", "Cont
 				self.displayGroup.imbuedSupport = nil
 			end
 		end
-		self.modFlag = true
 	end, true, true)
 	local function isImbuedEnabled() -- socketedIn must be set and the displayGroup must have an imbued, otherwise disable the imbued dropdown
 		return (self.displayGroup and self.displayGroup.slot and ((self.imbuedSupportBySlot[self.displayGroup.slot] and self.displayGroup.imbuedSupport) or not self.imbuedSupportBySlot[self.displayGroup.slot]))
@@ -254,8 +254,8 @@ local SkillsTabClass = newClass("SkillsTab", "UndoHandler", "ControlHost", "Cont
 		self.controls.imbuedSupport:SetText("")
 		self.displayGroup.imbuedSupport = nil
 		self.imbuedSupportBySlot[self.displayGroup.slot] = nil
+		self:AddUndoState()
 		self.build.buildFlag = true
-		self.modFlag = true
 	end)
 	self.controls.imbuedSupportClear.enabled = function()
 		return isImbuedEnabled()