Index: firmware/App/Modes/ModePreTreat.c
===================================================================
diff -u -ra25c11a9beb66c7292a184e3bc02a1a5ded5ed8a -r7076e6cdb254466b9bb119deaae70e7e6ab02015
--- firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision a25c11a9beb66c7292a184e3bc02a1a5ded5ed8a)
+++ firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 7076e6cdb254466b9bb119deaae70e7e6ab02015)
@@ -382,6 +382,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