Index: tst_Treatment_Adjustment_Ultrafiltration/test.py
===================================================================
diff -u -r867655840cada2d79fa5028c8fab68516ea59673 -rcb8d48033154f4b36a6a05b6f31bd10390e84ac7
--- tst_Treatment_Adjustment_Ultrafiltration/test.py	(.../test.py)	(revision 867655840cada2d79fa5028c8fab68516ea59673)
+++ tst_Treatment_Adjustment_Ultrafiltration/test.py	(.../test.py)	(revision cb8d48033154f4b36a6a05b6f31bd10390e84ac7)
@@ -355,7 +355,7 @@
     closeScreen() # Back to main
 
     # set the ultrafiltration to running state
-    HDSimulator.cmd_set_treatment_states_data(TXStates.TREATMENT_DIALYSIS_STATE_NOT, TXStates.UF_RUNNING_STATE, TXStates.SALINE_BOLUS_STATE_IDLE)
+    HDSimulator.cmd_set_treatment_states_data(TXStates.TREATMENT_DIALYSIS_STATE_NOT, TXStates.UF_RUNNING_STATE, TXStates.SALINE_BOLUS_STATE_IDLE, TXStates.HEPARIN_STATE_OFF)
 
     gotoScreen_UltrafiltrationAdjustment()