Index: firmware/App/Modes/ModePostTreat.c
===================================================================
diff -u -r5fcae5901d466789c329256f5532cf64be549c83 -r802a590448df6a6cdd949f9a433559d7be89c034
--- firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision 5fcae5901d466789c329256f5532cf64be549c83)
+++ firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision 802a590448df6a6cdd949f9a433559d7be89c034)
@@ -153,7 +153,7 @@
     signalBloodPumpHardStop();
     signalDialOutPumpHardStop();
     signalDialInPumpHardStop();
-    setCurrentSubState( NO_SUB_STATE );
+    setCurrentSubState( (U32)currentDrainReservoirState );
     collectTreatmentLogData();
 
     exitBloodLeakNormalState();