Index: tst_treatment_monitoring/test.py
===================================================================
diff -u -rff4b21e2f2457981d06cddbcae432c996bcaceb3 -r6bb43117bca2673c5de877f5b70b094da344418a
--- tst_treatment_monitoring/test.py	(.../test.py)	(revision ff4b21e2f2457981d06cddbcae432c996bcaceb3)
+++ tst_treatment_monitoring/test.py	(.../test.py)	(revision 6bb43117bca2673c5de877f5b70b094da344418a)
@@ -19,7 +19,7 @@
 from dialin.ui import utils
 from dialin.ui.hd_simulator import HDSimulator
 from configuration import config
-from dialin.common.hd_defs import HDOpModes, HDOpSubModes, TreatmentStates
+from dialin.common.hd_defs import HDOpModes, HDStandbyStates, TreatmentStates
 
 BLOOD_PRIMING_TARGET_VALUE  = 150
 
@@ -32,6 +32,7 @@
     @param: expected_blood_prime_value - expected blood prime value.    
     @return: N/A 
     """
+    utils.waitForGUI(0.5)
     expected_blood_prime_value = str(expected_blood_prime_value) +" "+config.LIQUID_UNIT
     test.compare(expected_blood_prime_value, waitForObject(names.o_target_value).text)
     
@@ -63,7 +64,7 @@
     @return: N/A
     """
     test.startSection("verification of treatment parameters")
-    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)