diff --git a/meshroom/ui/qml/Viewer/Viewer2D.qml b/meshroom/ui/qml/Viewer/Viewer2D.qml
index 83857cfa..b1912a38 100644
--- a/meshroom/ui/qml/Viewer/Viewer2D.qml
+++ b/meshroom/ui/qml/Viewer/Viewer2D.qml
@@ -1096,6 +1096,9 @@ FocusScope {
                         }
                         MaterialToolButton {
                             id: displayLensDistortionViewer
+
+                            property int numberChanges: null
+                            property bool previousChecked: null
                             property var activeNode: root.aliceVisionPluginAvailable && _reconstruction ? _reconstruction.activeNodes.get('sfmData').node : null
                             property bool isComputed: {
                                 if (!activeNode)
@@ -1127,6 +1130,22 @@ FocusScope {
                                     displayHDR.checked = true
                                 }
                             }
+
+                            onActiveNodeChanged: {
+                                numberChanges += 1
+                            }
+
+                            onEnabledChanged: {
+                                if (!enabled) {
+                                    previousChecked = checked
+                                    checked = false
+                                    numberChanges = 0
+                                }
+
+                                if (enabled && (numberChanges == 1) && previousChecked) {
+                                    checked = true
+                                }
+                            }
                         }
                         MaterialToolButton {
                             id: displayPanoramaViewer