Index: firmware/App/Modes/ModePostTreat.c
===================================================================
diff -u -r24b2fe72608344e67ef37234085d15ad5e4fcc37 -r5420b3e1f4260d8fbdce3f4f495f19f1a6f05b7e
--- firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision 24b2fe72608344e67ef37234085d15ad5e4fcc37)
+++ firmware/App/Modes/ModePostTreat.c	(.../ModePostTreat.c)	(revision 5420b3e1f4260d8fbdce3f4f495f19f1a6f05b7e)
@@ -573,7 +573,7 @@
     }
 
     // Drain has started and DG goes to re-circ mode flush lines state means drain completed
-    if ( ( TRUE == isDrainStarted ) && ( DG_GEN_IDLE_MODE_STATE_FLUSH_LINES == getDGSubMode() ) )
+    if ( ( TRUE == isDrainStarted ) && ( DG_GEN_IDLE_MODE_STATE_FLUSH_WATER == getDGSubMode() ) )
     {
         isDrainCompleted = TRUE;
     }
@@ -632,7 +632,7 @@
     }
 
     // Drain has started and DG goes to generation idle mode flush lines state means drain completed
-    if ( ( TRUE == isDrainStarted ) && ( DG_GEN_IDLE_MODE_STATE_FLUSH_LINES == getDGSubMode() ) )
+    if ( ( TRUE == isDrainStarted ) && ( DG_GEN_IDLE_MODE_STATE_FLUSH_WATER == getDGSubMode() ) )
     {
         isDrainCompleted = TRUE;
     }