Index: tst_create_custom_treatment/test.py
===================================================================
diff -u -r860dba59c921a0ae5b6f1a17f42d053bf13b63e3 -re8da75c7ba46f21fe99cc7400f3616a715c2ea5e
--- tst_create_custom_treatment/test.py	(.../test.py)	(revision 860dba59c921a0ae5b6f1a17f42d053bf13b63e3)
+++ tst_create_custom_treatment/test.py	(.../test.py)	(revision e8da75c7ba46f21fe99cc7400f3616a715c2ea5e)
@@ -613,15 +613,15 @@
 
     utils.waitForGUI(0.3)
     test.log("Verification of page step indicator from patient id page")
-    utility.verify_page_step_indicator(names.o_PreTreatmentBase_gridSteps_Grid_1, PRE_TREATMENT_STEP)
+    utility.verify_page_step_indicator(screen_obj=names.o_PreTreatmentBase_gridSteps_Grid_1, treatment_step=PRE_TREATMENT_STEP, treatment_screens=config.PRE_TREATMENT_SCREENS)
 
     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(names.o_PreTreatmentCreate_gridSteps_Grid, PRE_TREATMENT_STEP)
+    utility.verify_page_step_indicator(screen_obj=names.o_PreTreatmentCreate_gridSteps_Grid, treatment_step=PRE_TREATMENT_STEP, treatment_screens=config.PRE_TREATMENT_SCREENS)
     
 
     verify_create_custom_treatment_parameter()