Index: suite_leahi/tst_main_treatment/test.py
===================================================================
diff -u -r9eecbb5ba577cea331207beeac5d04e0a0e9bf0a -rd5b09ce1c343318a5ca04fe9de4e8b396563eec4
--- suite_leahi/tst_main_treatment/test.py	(.../test.py)	(revision 9eecbb5ba577cea331207beeac5d04e0a0e9bf0a)
+++ suite_leahi/tst_main_treatment/test.py	(.../test.py)	(revision d5b09ce1c343318a5ca04fe9de4e8b396563eec4)
@@ -44,7 +44,7 @@
     {"uf_volume": 1,  "uf_rate": 0.53, "vol_delivered": 0.37  },
     {"uf_volume": 10, "uf_rate": 5.05, "vol_delivered": 2.37  },
     {"uf_volume": 20, "uf_rate": 10.0, "vol_delivered": 5.0   },
-    {"uf_volume": 100,"uf_rate": 140.0,"vol_delivered": 156.0 },
+    # {"uf_volume": 100,"uf_rate": 140.0,"vol_delivered": 156.0 },
     {"uf_volume": 0,  "uf_rate": 0.0,  "vol_delivered": 0.0   }
     ]
 
@@ -213,7 +213,7 @@
     @return: N/A
     """
     test.compare(utility.get_object_from_names(names.o_treatmentUltrafiltration).visible, True, "UltraFiltration section is visible")
-    test.compare(waitForObjectExists(names.o_UF_Volume_Text).topText, "UF Volume", "UF Volume text verified")
+    test.compare(waitForObjectExists(names.o_UF_Volume_Text).topText, "UF Volume Goal", "UF Volume text verified")
     test.compare(waitForObjectExists(names.o_UF_Rate_Text).topText, "UF Rate", "UF Rate text verified")
    
 def start_test_treatment_ultrafiltration(uf_volume,uf_rate,vol_delivered):
@@ -378,7 +378,7 @@
                     0 ,
                     0 ,
                     0)
-    
+
     # verify main treatment screen
     test.verify(waitForObjectExists(names.mainTreatmentScreen), "In Main Treatment")
     
@@ -416,7 +416,7 @@
     for value in total_time_list:
         treatment_time_verification(value)
     #Check the treatment complete text message
-    test.compare(utility.get_object_from_names(names.o_treatmentTimeNotificationBarSmall).text, "Treatment Complete",  "Tx time completed text should be :" + str("Treatment Complete"))
+    test.compare(waitForObjectExists(names.o_treatmentHome_notification_NotificationBarSmall,2000).text, "Treatment Complete",  "Tx time completed text should be :" + str("Treatment Complete"))
     td.td_treatment_time( tx_duration_s       = 51 , 
                           tx_elapsed_s        = 5  , 
                           tx_remaining_s      = 6  )
@@ -430,7 +430,7 @@
                     0 ,
                     0 ,
                     0 )
-    test.compare(utility.get_object_from_names(names.o_treatmentTimeNotificationBarSmall).text, "Treatment Paused",  "Tx time paused text should be :" + str("Treatment Paused"))
+    test.compare(waitForObjectExists(names.o_treatmentHome_notification_NotificationBarSmall).text, "Treatment Paused",  "Tx time paused text should be :" + str("Treatment Paused"))
     test.endSection()
     
     #TX Parameter Set Points