Index: suite_leahi/tst_pre_treatment_foundation/test.py
===================================================================
diff -u -r1e3869b12486cff17bef668c69a6cb36cafff4a8 -rd4c6bf838a87529d60797460eb7242948f46f65b
--- suite_leahi/tst_pre_treatment_foundation/test.py	(.../test.py)	(revision 1e3869b12486cff17bef668c69a6cb36cafff4a8)
+++ suite_leahi/tst_pre_treatment_foundation/test.py	(.../test.py)	(revision d4c6bf838a87529d60797460eb7242948f46f65b)
@@ -86,9 +86,16 @@
     )
 
 
-def set_treatment_params(patient_id, blood_flow, dialysate_flow, duration_value, ultrafiltration_volume_value):
-    mouseClick(waitForObject(names.o_PreTreatmentCreate_pretreatmentPatientIDEntry_TextEntry))
-    waitForObject(names.o_PreTreatmentCreate_pretreatmentPatientIDEntry_TextEntry).text = patient_id
+def set_treatment_params(patient_id_num, blood_flow, dialysate_flow, duration_value, ultrafiltration_volume_value):
+    patient_id = waitForObject(names.o_PreTreatmentCreate_patientID_LabelUnitContainer)
+    patient_id_edit = utility.findObjectById(patient_id, "_editButton")
+    mouseClick(patient_id_edit)
+    mouseClick(waitForObject(names.o_pretreatmentPatientIDEntry_TextEntry))
+    waitForObject(names.o_pretreatmentPatientIDEntry_TextEntry).text = patient_id_num
+    mouseClick(waitForObject(names.o_secondaryPatientID_LabelUnitContainer))
+    waitForObject(names.o_secondaryPatientID_LabelUnitContainer).text ="Test1"
+    type(names.o_pretreatmentPatientIDEntry_TextEntry, "<Return>")
+    mouseClick(waitForObject(names.o_confirmButton_ConfirmButton))
     parent_obj = waitForObject(names.o_PreTreatmentCreate_leftColumn_Column, 3000)
     blood_flow_rate = utility.findObjectById(parent_obj, "_bloodFlowRate")
     utility.set_value_based_on_target(blood_flow_rate, blood_flow)