Index: firmware/App/Modes/ModePostTreat.c
===================================================================
diff -u -r8bd1ae47aa13a843aa8abd6321ddc050deacb4a6 -rc137d3c7cb17b0364d745e10ff6dbd1901eb1baa
--- firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision 8bd1ae47aa13a843aa8abd6321ddc050deacb4a6)
+++ firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision c137d3c7cb17b0364d745e10ff6dbd1901eb1baa)
@@ -116,9 +116,9 @@
  * post-treatment mode.
  * @details Inputs: none
  * @details Outputs: none
- * @return none
+ * @return initial state
  *************************************************************************/
-void transitionToPostTreatmentMode( void )
+U32 transitionToPostTreatmentMode( void )
 {
     initPostTreatmentMode();
 
@@ -144,6 +144,8 @@
     signalDialInPumpHardStop();
 
     collectTreatmentLogData();
+
+    return currentPostTreatmentState;
 }
 
 /*********************************************************************//**