Index: firmware/App/Modes/Dialysis.c
===================================================================
diff -u -r521a0569bf3097e9f7075d9c0e99dca9af3e6410 -re0e66d4c302c2aa43951ca316689ed4debbff1fb
--- firmware/App/Modes/Dialysis.c	(.../Dialysis.c)	(revision 521a0569bf3097e9f7075d9c0e99dca9af3e6410)
+++ firmware/App/Modes/Dialysis.c	(.../Dialysis.c)	(revision e0e66d4c302c2aa43951ca316689ed4debbff1fb)
@@ -246,9 +246,11 @@
 BOOL resumeUF( void )
 {
     BOOL result = FALSE;
+    TREATMENT_STATE_T trtState = getTreatmentState();
     OP_MODE currMode = getCurrentOperationMode();
 
-    if ( ( MODE_TREA == currMode ) && ( DIALYSIS_UF_STATE == currentDialysisState ) && ( UF_PAUSED_STATE == currentUFState ) )
+    if ( ( MODE_TREA == currMode ) && ( TREATMENT_DIALYSIS_STATE == trtState ) &&
+         ( DIALYSIS_UF_STATE == currentDialysisState ) && ( UF_PAUSED_STATE == currentUFState ) )
     {
         S32 inletPumpRate = getTargetDialInFlowRate();