Index: simulator/plugins/pretreatmentstates/loader.py
===================================================================
diff -u -r967521dff94783ce48b50abc8db6bcd67bb501bf -r9696f8b120608ac427f8b85d4f8ed2fd259e5d9e
--- simulator/plugins/pretreatmentstates/loader.py	(.../loader.py)	(revision 967521dff94783ce48b50abc8db6bcd67bb501bf)
+++ simulator/plugins/pretreatmentstates/loader.py	(.../loader.py)	(revision 9696f8b120608ac427f8b85d4f8ed2fd259e5d9e)
@@ -78,15 +78,15 @@
         initializes the widgets' properties
         :return: none
         """
-        self.tblSubMode.setCurrentCell(HD_Pre_Treatment_Mode_States.HD_PRE_TREATMENT_START_STATE.value, 0)
-        self.tblWaterSampleStates.setCurrentCell(HD_Pre_Treatment_Sample_Water_States.SAMPLE_WATER_SETUP_STATE.value, 0)
-        self.tblConsumablesSelfTestStates.setCurrentCell(HD_Pre_Treatment_Consumable_Self_Test_States.CONSUMABLE_SELF_TESTS_INSTALL_STATE.value, 0)
-        self.tblNoCartridgeSelfTestStates.setCurrentCell(HD_Pre_Treatment_No_Cart_Self_Tests_States.NO_CART_SELF_TESTS_START_STATE.value, 0)
-        self.tblInstallationStates.setCurrentCell(HD_Pre_Treatment_Cartridge_Install_States.CARTRIDGE_INSTALL_STATE.value, 0)
-        self.tblDrySelfTestStates.setCurrentCell(HD_Pre_Treatment_Dry_Self_Tests_States.DRY_SELF_TESTS_START_STATE.value, 0)
-        self.tblPrimeStates.setCurrentCell(HD_Pre_Treatment_Prime_States.HD_PRIME_START_STATE.value, 0)
-        self.tblRecirculateStates.setCurrentCell(HD_Pre_Treatment_Recirc_States.PRE_TREATMENT_RECIRC_STATE.value, 0)
-        self.tblPatientConnectionStates.setCurrentCell(HD_Pre_Treatment_Patient_Connection_States.PATIENT_CONNECTION_STATE.value, 0)
+        self.tblSubMode.setCurrentCell(PreTreatmentModeStates.HD_PRE_TREATMENT_START_STATE.value, 0)
+        self.tblWaterSampleStates.setCurrentCell(PreTreatmentSampleWaterStates.SAMPLE_WATER_SETUP_STATE.value, 0)
+        self.tblConsumablesSelfTestStates.setCurrentCell(PreTreatmentConsumableSelfTestStates.CONSUMABLE_SELF_TESTS_INSTALL_STATE.value, 0)
+        self.tblNoCartridgeSelfTestStates.setCurrentCell(PreTreatmentNoCartSelfTestsStates.NO_CART_SELF_TESTS_START_STATE.value, 0)
+        self.tblInstallationStates.setCurrentCell(PreTreatmentCartridgeInstallStates.CARTRIDGE_INSTALL_STATE.value, 0)
+        self.tblDrySelfTestStates.setCurrentCell(PreTreatmentDrySelfTestsStates.DRY_SELF_TESTS_START_STATE.value, 0)
+        self.tblPrimeStates.setCurrentCell(PreTreatmentPrimeStates.HD_PRIME_START_STATE.value, 0)
+        self.tblRecirculateStates.setCurrentCell(PreTreatmentRecircStates.PRE_TREATMENT_RECIRC_STATE.value, 0)
+        self.tblPatientConnectionStates.setCurrentCell(PreTreatmentPatientConnectionStates.PATIENT_CONNECTION_STATE.value, 0)
 
         self.tblSubMode.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)
         self.tblWaterSampleStates.horizontalHeaderItem(0).setTextAlignment(QtCore.Qt.AlignLeft)