Index: leahi_dialin/td/treatment_delivery.py
===================================================================
diff -u -r18c90a1b2b6c7339bdd192a2d2fac32f2b8a35df -rc5bfcf84ed942ca5841bf9de155aa0f495b4d28d
--- leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 18c90a1b2b6c7339bdd192a2d2fac32f2b8a35df)
+++ leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision c5bfcf84ed942ca5841bf9de155aa0f495b4d28d)
@@ -37,7 +37,7 @@
 
 from ..common.constants import NO_RESET
 from ..common.msg_defs import MsgIds, MsgFieldPositions, MsgFieldPositionsFWVersions
-from ..common.td_defs import TDOpModes
+from ..common.td_defs import td_enum_repository
 from ..common.override_templates import cmd_generic_broadcast_interval_override, cmd_generic_override
 from ..protocols.CAN import DenaliMessage, DenaliCanMessenger, DenaliChannels
 from ..utils.base import AbstractSubSystem, publish, LogManager
@@ -106,7 +106,7 @@
         self.td_debug_events_timestamp = 0.0
         self.ui_version_info_response_timestamp = 0.0
         self.td_version_response_timestamp = 0.0
-        self.td_operation_mode = TDOpModes.MODE_INIT.value
+        self.td_operation_mode = td_enum_repository.TDOpModes.MODE_INIT.value
         self.td_operation_sub_mode = 0
         self.td_logged_in = False
         self.td_set_logged_in_status(False)
@@ -346,7 +346,7 @@
             channel_id = DenaliChannels.dialin_to_td_ch_id,
             msg_id = MsgIds.MSG_ID_TD_OP_MODE_OVERRIDE_REQUEST,
             entity_name = 'TD Operation Mode',
-            override_text = TDOpModes(new_mode).name,
+            override_text = td_enum_repository.TDOpModes(new_mode).name,
             logger = self.logger,
             can_interface = self.can_interface)