Index: tst_instructions_poc/test.py
===================================================================
diff -u -r66899cebdc7c1bbde4556d947f87a671a782ad4e -r20ef0b22dacadabdc0c496c3e7e73657e06ed3b1
--- tst_instructions_poc/test.py	(.../test.py)	(revision 66899cebdc7c1bbde4556d947f87a671a782ad4e)
+++ tst_instructions_poc/test.py	(.../test.py)	(revision 20ef0b22dacadabdc0c496c3e7e73657e06ed3b1)
@@ -19,7 +19,7 @@
 from dialin.ui.hd_simulator import HDSimulator
 from configuration import config
 from configuration import utility
-from dialin.common.hd_defs import HDOpModes, HDOpSubModes, PreTreatmentSubModes, PreTreatmentSampleWaterStates, PreTreatmentConsumableSelfTestStates, PreTreatmentNoCartSelfTestsStates, PreTreatmentCartridgeInstallStates, PreTreatmentDrySelfTestsStates,  PreTreatmentPrimeStates, PreTreatmentRecircStates, PreTreatmentPatientConnectionStates, TreatmentStates, UFStates, SalineBolusStates, HeparinStates, TreatmentRinsebackStates, PreTreatmentRecircStates, TreatmentBloodPrimeStates, TreatmentEndStates, TreatmentStopStates, TreatmentDialysisStates,PostTreatmentStates
+from dialin.common.hd_defs import HDOpModes, HDOpSubModes, HDInitStates, PreTreatmentSubModes, PreTreatmentSampleWaterStates, PreTreatmentConsumableSelfTestStates, PreTreatmentNoCartSelfTestsStates, PreTreatmentCartridgeInstallStates, PreTreatmentDrySelfTestsStates,  PreTreatmentPrimeStates, PreTreatmentRecircStates, PreTreatmentPatientConnectionStates, TreatmentStates, UFStates, SalineBolusStates, HeparinStates, TreatmentRinsebackStates, PreTreatmentRecircStates, TreatmentBloodPrimeStates, TreatmentEndStates, TreatmentStopStates, TreatmentDialysisStates
 
 hd_simulator = HDSimulator()
 ven_low = 40
@@ -412,7 +412,7 @@
     test.endSection()
      
     test.startSection("Navigating to Post Treatment Disconnection screen")
-    hd_simulator.cmd_send_hd_operation_mode(op_mode = HDOpModes.MODE_POST.value , sub_mode = PostTreatmentStates.HD_POST_TREATMENT_PATIENT_DISCONNECTION_STATE.value)
+    hd_simulator.cmd_send_hd_operation_mode(op_mode = HDOpModes.MODE_POST.value , sub_mode = HDInitStates.POST_STATE_START.value)
     index = verify_instruction_and_image("Patient Disconnection",index = index)
     mouseClick(waitForObject(names.o_confirm_button_patient_disconnection))
     mouseClick(waitForObject(names.o_next_button_review))