Index: TD_TreatmentStates/loader.py
===================================================================
diff -u -r2804a0aa7699197ae2cbaf2d68cd1f9dd30172dd -r6fe1ca7c7d9678cf71f8fff8b3bb7988b8231207
--- TD_TreatmentStates/loader.py	(.../loader.py)	(revision 2804a0aa7699197ae2cbaf2d68cd1f9dd30172dd)
+++ TD_TreatmentStates/loader.py	(.../loader.py)	(revision 6fe1ca7c7d9678cf71f8fff8b3bb7988b8231207)
@@ -10,7 +10,7 @@
 from engine.dynamicloader   import DynamicLoader
 
 # plugin specific
-from leahi_dialin.common.td_defs import TDTreatmentStates, TDDialysisStates
+from leahi_dialin.common.td_defs import TDTreatmentStates, TDTreaDialysisStates
 
 # hd Simulator
 from leahi_dialin.ui.td_messaging import TD_Messaging
@@ -71,9 +71,9 @@
         initializes the widgets' properties
         :return: none
         """
-        self.tblSubMode            .setCurrentCell(TDTreatmentStates.TREATMENT_START_STATE   .value  , 0)
+        self.tblSubMode             .setCurrentCell(TDTreatmentStates.TREATMENT_START_STATE.value, 0)
         self.tblBloodPrimeStates   .setCurrentCell(0                                        , 0)
-        self.tblDialysisStates     .setCurrentCell(TDDialysisStates.DIALYSIS_UF_STATE     .value  , 0)
+        self.tblDialysisStates      .setCurrentCell(TDTreaDialysisStates.DIALYSIS_UF_STATE.value, 0)
         self.tblUFStates           .setCurrentCell(0                                        , 0)
         self.tblTreatmentStopStates.setCurrentCell(0                                        , 0)
         self.tblRinsebackStates    .setCurrentCell(0                                        , 0)