Index: firmware/App/Modes/ModePreTreat.c
===================================================================
diff -u -r8466e63f95f65a3ffb18c3af85ac99328e41167b -r2e21b8a4e52e0d1f94389acf424b5033362b63f1
--- firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 8466e63f95f65a3ffb18c3af85ac99328e41167b)
+++ firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 2e21b8a4e52e0d1f94389acf424b5033362b63f1)
@@ -376,6 +376,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