Index: tst_Treatment_Adjustment_Ultrafiltration/test.py
===================================================================
diff -u -r24485172848ae32d15cf665b2ef024129dec6ca2 -r316fd2f51b0fa89c0f69949d79ef246e1e4bc4ed
--- tst_Treatment_Adjustment_Ultrafiltration/test.py	(.../test.py)	(revision 24485172848ae32d15cf665b2ef024129dec6ca2)
+++ tst_Treatment_Adjustment_Ultrafiltration/test.py	(.../test.py)	(revision 316fd2f51b0fa89c0f69949d79ef246e1e4bc4ed)
@@ -1,16 +1,19 @@
 # -*- coding: utf-8 -*-
-
-##
-# Copyright (c) 2019-2020 Diality Inc. - All Rights Reserved.
-# copyright
-#          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    tst_Treatment_Ultrafiltration
-# date    2020/06/17
-# author  Behrouz NematiPour
+# Copyright (c) 2020-2025 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    test.py
+#
+# @author  (last)      Behrouz NematiPour
+# @date    (last)      07-Jan-2021
+# @author  (original)  Behrouz NemaiPour
+# @date    (original)  17-Jun-2020
+#
+############################################################################
 
 import names
 from dialin.squish import utils, denaliMessages
@@ -241,7 +244,7 @@
 
 
 def testScreen_Confirm_Ultrafiltration():
-    #[Confrim Screen]
+    #[Confirm Screen]
     test.compare(    waitForObjectExists(names.o_treatmentAdjustmentUltrafiltrationConfirm_Screen).visible, True)
     test.compare(str(waitForObjectExists(names.o_treatmentAdjustmentUltrafiltrationConfirm_PausedText).text), "Ultrafiltration is paused")
     test.compare(str(waitForObjectExists(names.o_treatmentAdjustmentUltrafiltrationConfirm_ConfirmButton).text.text), "CONFIRM")
@@ -353,7 +356,7 @@
     closeScreen() # Back to main
 
     # set the ultrafiltration to running state
-    denaliMessages.setTreatmentStatesData(txStates.TREATMENT_DIALYSIS_STATE_NOT, txStates.UF_RUNNING_STATE, txStates.SALINE_BOLUS_STATE_IDLE )
+    denaliMessages.setTreatmentStatesData(txStates.TREATMENT_DIALYSIS_STATE_NOT, txStates.UF_RUNNING_STATE, txStates.SALINE_BOLUS_STATE_IDLE, txStates.HEPARIN_STATE_OFF)
 
     gotoScreen_UltrafiltrationAdjustment()