Index: firmware/App/Modes/ModePreTreat.c
===================================================================
diff -u -r8bd1ae47aa13a843aa8abd6321ddc050deacb4a6 -rc137d3c7cb17b0364d745e10ff6dbd1901eb1baa
--- firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 8bd1ae47aa13a843aa8abd6321ddc050deacb4a6)
+++ firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision c137d3c7cb17b0364d745e10ff6dbd1901eb1baa)
@@ -151,9 +151,9 @@
  * pre-treatment mode.
  * @details Inputs: none
  * @details Outputs: none
- * @return none
+ * @return initial state
 *************************************************************************/
-void transitionToPreTreatmentMode( void )
+U32 transitionToPreTreatmentMode( void )
 {
     F32 const trimmerHeaterTemp = getTreatmentParameterF32( TREATMENT_PARAM_DIALYSATE_TEMPERATURE );
     F32 const primaryHeaterTemp = trimmerHeaterTemp + PRIMARY_HEATER_TARGET_TEMP_OFFSET;
@@ -166,6 +166,8 @@
     setAlarmUserActionEnabled( ALARM_USER_ACTION_RESUME, TRUE );
     setAlarmUserActionEnabled( ALARM_USER_ACTION_RINSEBACK, FALSE );
     setAlarmUserActionEnabled( ALARM_USER_ACTION_END_TREATMENT, TRUE );
+
+    return currentPreTreatmentState;
 }
 
 /*********************************************************************//**