Index: TD_End_Treatment/loader.py
===================================================================
diff -u -rb8d218923616a2a64366c45172a83184914415bd -ra7779af44ced3be6e30e4d4d67125d171a641d09
--- TD_End_Treatment/loader.py	(.../loader.py)	(revision b8d218923616a2a64366c45172a83184914415bd)
+++ TD_End_Treatment/loader.py	(.../loader.py)	(revision a7779af44ced3be6e30e4d4d67125d171a641d09)
@@ -15,7 +15,6 @@
 
 # plugin specific
 from leahi_dialin.common.msg_defs import MsgIds
-from leahi_dialin.common.td_defs  import TDTreatmentStates
 from leahi_dialin.protocols       import CAN
 
 # hd Simulator
@@ -32,7 +31,7 @@
         super().__init__(os.path.dirname(__file__))
 
         if self.can_interface is not None:
-            channel_id = CAN.DenaliChannels.ui_to_td_ch_id
+            channel_id = CAN.CanChannels.ui_to_td_ch_id
             message_id = MsgIds.MSG_ID_TD_END_TREATMENT_REQUEST.value
             self.can_interface.register_receiving_publication_function(channel_id,
                                                                        message_id,