Index: leahi_dialin/td/treatment_delivery.py
===================================================================
diff -u -r18c90a1b2b6c7339bdd192a2d2fac32f2b8a35df -r38e8dd31728056dbb7f9304c98ab16d7147b75a2
--- leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 18c90a1b2b6c7339bdd192a2d2fac32f2b8a35df)
+++ leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 38e8dd31728056dbb7f9304c98ab16d7147b75a2)
@@ -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)