Index: tst_create_custom_treatment/test.py
===================================================================
diff -u -rb6887f320c7cda9c70b8f47cb44ebe6fc0c892f0 -rcea724e0b33dcd446f230ab23d6aa3729e6f86fd
--- tst_create_custom_treatment/test.py	(.../test.py)	(revision b6887f320c7cda9c70b8f47cb44ebe6fc0c892f0)
+++ tst_create_custom_treatment/test.py	(.../test.py)	(revision cea724e0b33dcd446f230ab23d6aa3729e6f86fd)
@@ -594,16 +594,18 @@
     
     hd_simulator.cmd_send_hd_operation_mode(4, 0)
 
+    utils.waitForGUI(0.3)
     test.log("Verification of page step indicator from patient id page")
-    utility.verify_page_step_indicator(PRE_TREATMENT_STEP, names.o_PreTreatmentBase_nextStepsBullet_StepBullet)
+    utility.verify_page_step_indicator(names.o_PreTreatmentCreateStack_PreTreatmentCreateStack, PRE_TREATMENT_STEP)
 
     mouseClick(waitForObject(names.o_input_patient_id))
     type(waitForObject(names.o_input_patient_id), config.PATIENT_ID)
     test.compare(waitForObject(names.o_PreTreatmentBase_Patient_ID_Text).text, config.PATIENT_ID_TEXT, "Title text should be -> "+str(config.PATIENT_ID_TEXT))
     mouseClick(waitForObject(names.o_confirm_button))
     
     test.log("Verification of page step indicator from create custom treatment page")
-    utility.verify_page_step_indicator(PRE_TREATMENT_STEP, names.o_PreTreatmentCreate_nextStepsBullet_StepBullet)
+    utils.waitForGUI(0.3)
+    utility.verify_page_step_indicator(names.o_PreTreatmentCreateStack_PreTreatmentCreateStack, PRE_TREATMENT_STEP)
 
     verify_create_custom_treatment_parameter()