Index: tst_main_treatment_ultrafiltration/test.py
===================================================================
diff -u -raef5d2dd35d5e35f1f68dc937a5edba1d30e2351 -r6bb43117bca2673c5de877f5b70b094da344418a
--- tst_main_treatment_ultrafiltration/test.py	(.../test.py)	(revision aef5d2dd35d5e35f1f68dc937a5edba1d30e2351)
+++ tst_main_treatment_ultrafiltration/test.py	(.../test.py)	(revision 6bb43117bca2673c5de877f5b70b094da344418a)
@@ -20,7 +20,7 @@
 from dialin.ui import utils
 from dialin.common.ui_defs import TXStates as uf_states
 from dialin.ui.hd_simulator import HDSimulator
-from dialin.common.hd_defs import HDOpSubModes, TreatmentStates
+from dialin.common.hd_defs import HDStandbyStates, TreatmentStates
 from configuration import config
 
 hd_simulator    = HDSimulator()
@@ -258,6 +258,7 @@
 
     for rejection in range(1, config.NUM_OF_REQUEST_REJECT_REASONS + 1):   
         hd_simulator.cmd_set_treatment_adjust_ultrafiltration_rejected(reason = rejection, state = uf_states)
+        utils.waitForGUI(0.5)
         rejection_message = waitForObjectExists(names.o_notificationbar_response)
         test.compare(rejection_message.text, config.REJECTION_REASON[rejection], "expected rejection {msg} displayed".format(msg=config.REJECTION_REASON[rejection]))
     
@@ -321,7 +322,7 @@
     startApplication(config.AUT_NAME)
     
     #navigate to dialysis state
-    hd_simulator.cmd_set_treatment_states_data(sub_mode=HDOpSubModes.STANDBY_WAIT_FOR_DISINFECT_STATE.value, uf_state=TreatmentStates.TREATMENT_START_STATE.value, saline_state=TreatmentStates.TREATMENT_START_STATE.value, heparin_state=TreatmentStates.TREATMENT_START_STATE.value,
+    hd_simulator.cmd_set_treatment_states_data(sub_mode=HDStandbyStates.STANDBY_WAIT_FOR_DISINFECT_STATE.value, uf_state=TreatmentStates.TREATMENT_START_STATE.value, saline_state=TreatmentStates.TREATMENT_START_STATE.value, heparin_state=TreatmentStates.TREATMENT_START_STATE.value,
                                       rinseback_state=TreatmentStates.TREATMENT_START_STATE.value, recirculate_state=TreatmentStates.TREATMENT_START_STATE.value, blood_prime_state=TreatmentStates.TREATMENT_START_STATE.value,
                                       treatment_end_state=TreatmentStates.TREATMENT_START_STATE.value, treatment_stop_state=TreatmentStates.TREATMENT_START_STATE.value, dialysis_state=TreatmentStates.TREATMENT_START_STATE.value)