Index: sources/gui/qml/pages/treatment/TreatmentHome.qml
===================================================================
diff -u -rca05e940ea4a0c567acb05c4c8777dea079b0f08 -r56b0b13aa9f9e7ae201bce1843015fe5b4b43d2c
--- sources/gui/qml/pages/treatment/TreatmentHome.qml	(.../TreatmentHome.qml)	(revision ca05e940ea4a0c567acb05c4c8777dea079b0f08)
+++ sources/gui/qml/pages/treatment/TreatmentHome.qml	(.../TreatmentHome.qml)	(revision 56b0b13aa9f9e7ae201bce1843015fe5b4b43d2c)
@@ -65,7 +65,6 @@
                 width       : (cellWidth * 2) - (Variables.defaultMargin * 2)
                 height      : cellHeight
                 onClicked   : sectionTimeClicked()
-                editEnabled : ! isTreatmentEnd      // LEAHI-PRS-376
             }
 
             TreatmentSaline { id: _treatmentSaline
@@ -106,12 +105,11 @@
                 objectName          : "treatmentFlows"
                 width               : cellWidth * 4
                 height              : cellHeight
-                header.editEnabled  : ! isTreatmentEnd      // LEAHI-PRS-376
-                header.lockEnabled  : ! isTreatmentEnd
-
+                header.setEditActive: ! isTreatmentEnd  // LEAHI-PRS-376
+                editEnabled         : ! _lockDialog.isLocked
                 onEditClicked   : sectionFlowClicked()
                 onLockClicked   :  {
-                    _lockDialog.isLocked = editEnabled
+                    _lockDialog.isLocked =  editEnabled
                     _lockDialog.showDialog (
                                 editEnabled ? qsTr("Treatment Parameters Unlocked") :
                                               qsTr("Treatment Parameters Locked"),