Index: TD_TreatmentStates/loader.py
===================================================================
diff -u -rb8705aa1e0402a49cc266c642facb833888116bd -rdc978465a0b39f201401a4b60bb0bb70f16c728b
--- TD_TreatmentStates/loader.py	(.../loader.py)	(revision b8705aa1e0402a49cc266c642facb833888116bd)
+++ TD_TreatmentStates/loader.py	(.../loader.py)	(revision dc978465a0b39f201401a4b60bb0bb70f16c728b)
@@ -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
@@ -73,7 +73,7 @@
         """
         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)