Index: leahi_dialin/td/treatment_delivery.py
===================================================================
diff -u -r76bd18ef10a248d1375dbca242ed3123281041b2 -rd27c40205e18118f4689f62463a266da1f91dba6
--- leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision 76bd18ef10a248d1375dbca242ed3123281041b2)
+++ leahi_dialin/td/treatment_delivery.py	(.../treatment_delivery.py)	(revision d27c40205e18118f4689f62463a266da1f91dba6)
@@ -36,7 +36,7 @@
 from .proxies.ui_proxy import UIProxy
 
 from ..common.msg_defs import MsgIds, MsgFieldPositions, MsgFieldPositionsFWVersions
-from ..common.td_defs import TDOpModes
+from ..common.td_defs import td_enum_repository
 from ..protocols.CAN import DenaliMessage, DenaliCanMessenger, DenaliChannels
 from ..utils.base import AbstractSubSystem, publish, LogManager
 from ..utils.checks import check_broadcast_interval_override_ms
@@ -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)