Index: firmware/App/Modes/ModeTreatment.c
===================================================================
diff -u -r3227fa2625279833f22926c981309b7ae11bdfca -r8bd1ae47aa13a843aa8abd6321ddc050deacb4a6
--- firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 3227fa2625279833f22926c981309b7ae11bdfca)
+++ firmware/App/Modes/ModeTreatment.c	(.../ModeTreatment.c)	(revision 8bd1ae47aa13a843aa8abd6321ddc050deacb4a6)
@@ -1,17 +1,17 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2019-2021 Diality Inc. - All Rights Reserved.
 *
 * 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 ModeTreatment.c
+* @file    ModeTreatment.c
 *
-* @author (last) Sean Nash
-* @date (last) 12-Oct-2020
+* @author  (last)      Quang Nguyen
+* @date    (last)      27-Aug-2021
 *
-* @author (original) Dara Navaei
-* @date (original) 05-Nov-2019
+* @author  (original)  Dara Navaei
+* @date    (original)  05-Nov-2019
 *
 ***************************************************************************/
 
@@ -606,6 +606,11 @@
         activateAlarmNoData( ALARM_ID_TREATMENT_STOPPED_BY_USER );
     }
 
+    if ( currentTreatmentState != TREATMENT_END_STATE )
+    {
+        checkDialysateTemperature();
+    }
+
     // Treatment mode state machine
     switch ( currentTreatmentState )
     {