Index: firmware/App/Modes/TreatmentEnd.c
===================================================================
diff -u -rce64b7e15824641a6a6465c28eec5d8ec593f110 -ra1a592eaed002ee7ef84e56fb7864b8d40b14661
--- firmware/App/Modes/TreatmentEnd.c	(.../TreatmentEnd.c)	(revision ce64b7e15824641a6a6465c28eec5d8ec593f110)
+++ firmware/App/Modes/TreatmentEnd.c	(.../TreatmentEnd.c)	(revision a1a592eaed002ee7ef84e56fb7864b8d40b14661)
@@ -114,7 +114,7 @@
     // Set user alarm recovery actions allowed in this sub-mode
     setAlarmUserActionEnabled( ALARM_USER_ACTION_RINSEBACK, TRUE );
     setAlarmUserActionEnabled( ALARM_USER_ACTION_END_TREATMENT, TRUE );
-
+    setCurrentSubState( (U32)treatmentEndState );
     // Set valves, pumps and heater
     setValvePosition( VDI, VALVE_POSITION_C_CLOSE );
     setValvePosition( VDO, VALVE_POSITION_C_CLOSE );