Index: suite_leahi/tst_end_of_treatment/test.py
===================================================================
diff -u -r7fa3503b0eeaf8e05f4f885d7af4ea2947d90e0a -r773bdca9f28d573177bb59389312e1a24b2f138c
--- suite_leahi/tst_end_of_treatment/test.py	(.../test.py)	(revision 7fa3503b0eeaf8e05f4f885d7af4ea2947d90e0a)
+++ suite_leahi/tst_end_of_treatment/test.py	(.../test.py)	(revision 773bdca9f28d573177bb59389312e1a24b2f138c)
@@ -118,16 +118,7 @@
     heart_rate = waitForObjectExists(names.o_treatmentHome_heartBeat_LabelValue).topText
     test.compare(config.MIN_HEARTRATE, heart_rate,"Heart Rate value should be ->"+str(config.MIN_HEARTRATE))
     ultrafiltration_edit = waitForObjectExists(names.o_editButton_ultrafiltration_IconButton)
-    test.compare(ultrafiltration_edit.enabled,True,"Ultrafiltration edit button is enabled")
-    mouseClick(waitForObject(names.o_editButton_ultrafiltration_IconButton))
-    #check isolated,pause uf and edit uf button state
-    isolateduf_button = waitForObjectExists(names.o_isolatedUfButton_TouchRect)
-    test.compare(isolateduf_button.enabled,True,"Isolated UF button is enabled")
-    edituf_button =waitForObjectExists(names.o_editButton_TouchRect)
-    test.compare(not edituf_button.enabled,True,"Edit UF button is disabled")
-    pauseuf_button = waitForObjectExists(names.o_pauseResumeButton_TouchRect)
-    test.compare(not pauseuf_button.enabled,True,"Pause UF button is disabled")
-    mouseClick(names.o_image_Image)
+    test.compare(ultrafiltration_edit.enabled,False,"Ultrafiltration edit button is enabled")
     treatment_parameters_edit = waitForObjectExists(names.o_treatmentHome_editButton_IconButton_2)
     test.compare(treatment_parameters_edit.enabled,True,"Treatment parameters edit button is disabled")
     td.td_treatment_set_points    (blood_flow     = 51,