diff --git a/Source/mapping_editors/BatchToolsColourControls.cpp b/Source/mapping_editors/BatchToolsColourControls.cpp
index 46884be..1bc03ad 100644
--- a/Source/mapping_editors/BatchToolsColourControls.cpp
+++ b/Source/mapping_editors/BatchToolsColourControls.cpp
@@ -239,6 +239,10 @@ void BatchToolsColourControls::handleStatePropertyChange(juce::ValueTree stateIn
 
 void BatchToolsColourControls::layoutWasUpdated()
 {
-    if (!getHasChangesToSave())
+    auto desc = undoManager->getUndoDescription();
+    if (!desc.contains("BatchColour"))
         *lastSavedLayout = *getMappingData();
+    // if (!getHasChangesToSave())
+    //     *lastSavedLayout = *getMappingData();
+
 }