Index: tst_pre_treatment_consumables/test.py
===================================================================
diff -u -r6bb43117bca2673c5de877f5b70b094da344418a -r84c1284ecff1f30169d50afb688be18934eb8506
--- tst_pre_treatment_consumables/test.py	(.../test.py)	(revision 6bb43117bca2673c5de877f5b70b094da344418a)
+++ tst_pre_treatment_consumables/test.py	(.../test.py)	(revision 84c1284ecff1f30169d50afb688be18934eb8506)
@@ -23,11 +23,11 @@
 from dialin.ui import utils
 from configuration import config, utility
 from dialin.ui.hd_simulator import HDSimulator
-from dialin.common.hd_defs import HDOpModes, HDStandbyStates, PreTreatmentSampleWaterStates, PreTreatmentConsumableSelfTestStates, PreTreatmentNoCartSelfTestStates, PreTreatmentCartridgeInstallStates, PreTreatmentDrySelfTestsStates,  PreTreatmentPrimeStates, PreTreatmentRecircStates, PreTreatmentPatientConnectionStates, PreTreatmentModeStates
+from dialin.common.hd_defs import HDOpModes, HDStandbyStates, PreTreatmentSampleWaterStates, PreTreatmentConsumableSelfTestStates, PreTreatmentNoCartSelfTestStates, PreTreatmentCartridgeInstallStates, PreTreatmentDrySelfTestsStates,  PreTreatmentPrimeStates, PreTreatmentRecircStates, PreTreatmentPatientConnectionStates, PreTreatmentSubModes
 
 hd_simulator  = HDSimulator()
 
-PRE_TREATMENT_STEP = PreTreatmentModeStates.HD_PRE_TREATMENT_SELF_TEST_CONSUMABLE_STATE.value
+PRE_TREATMENT_STEP = PreTreatmentSubModes.HD_PRE_TREATMENT_SELF_TEST_CONSUMABLE_STATE.value
 SCREEN_OBJ= names.o_PreTreatmentStack_preTreatmentConsumablesStack_PreTreatmentConsumablesStack