Index: firmware/App/Modes/ModePreTreat.c
===================================================================
diff -u -r954107454f4c5d4cfbdfaefda5cc1cd15b1e18ac -r8bd1ae47aa13a843aa8abd6321ddc050deacb4a6
--- firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 954107454f4c5d4cfbdfaefda5cc1cd15b1e18ac)
+++ firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 8bd1ae47aa13a843aa8abd6321ddc050deacb4a6)
@@ -5,13 +5,13 @@
 * THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
 * WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
 *
-* @file ModePreTreat.c
+* @file    ModePreTreat.c
 *
-* @author (last) Sean Nash
-* @date (last) 06-Oct-2020
+* @author  (last)      Quang Nguyen
+* @date    (last)      24-Aug-2021
 *
-* @author (original) Dara Navaei
-* @date (original) 05-Nov-2019
+* @author  (original)  Dara Navaei
+* @date    (original)  05-Nov-2019
 *
 ***************************************************************************/
 
@@ -388,6 +388,10 @@
         case ALARM_ACTION_END_TREATMENT:
             if ( HD_PRE_TREATMENT_PRIME_STATE > currentPreTreatmentState )
             {
+                if ( HD_PRE_TREATMENT_WATER_SAMPLE_STATE == currentPreTreatmentState )
+                {
+                    cmdDGSampleWater( SAMPLE_WATER_CMD_END );
+                }
                 requestNewOperationMode( MODE_STAN );
             }
             else