Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -r8bd1ae47aa13a843aa8abd6321ddc050deacb4a6 -rc137d3c7cb17b0364d745e10ff6dbd1901eb1baa
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 8bd1ae47aa13a843aa8abd6321ddc050deacb4a6)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision c137d3c7cb17b0364d745e10ff6dbd1901eb1baa)
@@ -228,9 +228,9 @@
  * The transitionToTreatmentMode function prepares for transition to treatment mode.
  * @details Inputs: none
  * @details Outputs:
- * @return none
+ * @return initial state
  *************************************************************************/
-void transitionToTreatmentMode( void )
+U32 transitionToTreatmentMode( void )
 {
     // Initialize treatment mode each time we transition to it
     initTreatmentMode();
@@ -242,6 +242,8 @@
     initRinseback();
     initTreatmentRecirc();
     initTreatmentEnd();
+
+    return currentTreatmentState;
 }
 
 /*********************************************************************//**