Index: firmware/App/Modes/ModePreTreat.c
===================================================================
diff -u -r518745cc77b07b0a6aa35d5530255d2bcc8b1090 -rdd1c45cd9a612aaf63de14ffddecab573ead6095
--- firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision 518745cc77b07b0a6aa35d5530255d2bcc8b1090)
+++ firmware/App/Modes/ModePreTreat.c	(.../ModePreTreat.c)	(revision dd1c45cd9a612aaf63de14ffddecab573ead6095)
@@ -365,6 +365,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