Index: firmware/App/Modes/OperationModes.c
===================================================================
diff -u -r4760b847bd262dfc693b3d3458864c05b1c816d7 -rce247681f11b5e164e00992fb73ba5c595d349aa
--- firmware/App/Modes/OperationModes.c	(.../OperationModes.c)	(revision 4760b847bd262dfc693b3d3458864c05b1c816d7)
+++ firmware/App/Modes/OperationModes.c	(.../OperationModes.c)	(revision ce247681f11b5e164e00992fb73ba5c595d349aa)
@@ -7,8 +7,8 @@
 *
 * @file    OperationModes.c
 *
-* @author  (last)      Dara Navaei
-* @date    (last)      03-May-2022
+* @author  (last)      Darren Cox
+* @date    (last)      25-Aug-2022
 *
 * @author  (original)  Dara Navaei
 * @date    (original)  05-Nov-2019
@@ -140,7 +140,7 @@
         transitionToNewOperationMode( newMode );
         currentMode = newMode;
 
-        if ( MODE_TREA == currentMode )
+        if ( MODE_TREA == lastMode )
         {
             // If the current mode is treatment but transitioning to another mode has been requested (including transitioning to fault mode)
             // it means the treatment is done. Get the elapsed time since the beginning of the treatment and convert it to hours to be written