Index: leahi_dialin/td/modules/air_trap.py
===================================================================
diff -u -r43fec7787f9a4115d19deeb5cbc1d7bbf23134fe -rd27c40205e18118f4689f62463a266da1f91dba6
--- leahi_dialin/td/modules/air_trap.py	(.../air_trap.py)	(revision 43fec7787f9a4115d19deeb5cbc1d7bbf23134fe)
+++ leahi_dialin/td/modules/air_trap.py	(.../air_trap.py)	(revision d27c40205e18118f4689f62463a266da1f91dba6)
@@ -19,7 +19,7 @@
 
 from .constants import RESET, NO_RESET
 from leahi_dialin.common.msg_defs import MsgIds, MsgFieldPositions
-from leahi_dialin.common.td_defs import TDAirTrapValves, TDAirTrapLevelSensors, TDAirTrapLevelSensorsAttributes, TDAirTrapState
+from leahi_dialin.common.td_defs import td_enum_repository
 from leahi_dialin.protocols.CAN import DenaliMessage, DenaliChannels
 from leahi_dialin.utils.base import AbstractSubSystem, publish, DialinEnum
 from leahi_dialin.utils.checks import check_broadcast_interval_override_ms
@@ -53,18 +53,18 @@
                                                                        self._handler_air_trap_sync)
 
         self.air_trap_level_sensors = {
-            TDAirTrapLevelSensors.H17_LEVEL.name: {
-                TDAirTrapLevelSensorsAttributes.LEVEL.name: self.AIR_DETECTED_AT_LEVEL,
-                TDAirTrapLevelSensorsAttributes.RAW.name: self.AIR_DETECTED_AT_LEVEL
+            td_enum_repository.TDAirTrapLevelSensorNames.H17_LEVEL.name: {
+                td_enum_repository.TDAirTrapLevelSensorAttributes.LEVEL.name: self.AIR_DETECTED_AT_LEVEL,
+                td_enum_repository.TDAirTrapLevelSensorAttributes.RAW.name: self.AIR_DETECTED_AT_LEVEL
             },
-            TDAirTrapLevelSensors.H16_LEVEL.name: {
-                TDAirTrapLevelSensorsAttributes.LEVEL.name: self.AIR_DETECTED_AT_LEVEL,
-                TDAirTrapLevelSensorsAttributes.RAW.name: self.AIR_DETECTED_AT_LEVEL
+            td_enum_repository.TDAirTrapLevelSensorNames.H16_LEVEL.name: {
+                td_enum_repository.TDAirTrapLevelSensorAttributes.LEVEL.name: self.AIR_DETECTED_AT_LEVEL,
+                td_enum_repository.TDAirTrapLevelSensorAttributes.RAW.name: self.AIR_DETECTED_AT_LEVEL
             }
         }
         self.air_trap_valve_states = {
-            TDAirTrapValves.H13_VALV.name: 0,
-            TDAirTrapValves.H20_VALV.name: 0
+            td_enum_repository.TDAirTrapValveNames.H13_VALV.name: 0,
+            td_enum_repository.TDAirTrapValveNames.H20_VALV.name: 0
         }
         self.air_control = 0
         self.td_air_trap_timestamp = 0.0
@@ -98,12 +98,12 @@
         controlling = struct.unpack('i', bytearray(
             message['message'][MsgFieldPositions.START_POS_FIELD_7:MsgFieldPositions.END_POS_FIELD_7]))
 
-        self.air_trap_level_sensors[TDAirTrapLevelSensors.H17_LEVEL.name][TDAirTrapLevelSensorsAttributes.LEVEL.name] = lower[0]
-        self.air_trap_level_sensors[TDAirTrapLevelSensors.H16_LEVEL.name][TDAirTrapLevelSensorsAttributes.LEVEL.name] = upper[0]
-        self.air_trap_level_sensors[TDAirTrapLevelSensors.H17_LEVEL.name][TDAirTrapLevelSensorsAttributes.RAW.name] = raw_lower[0]
-        self.air_trap_level_sensors[TDAirTrapLevelSensors.H16_LEVEL.name][TDAirTrapLevelSensorsAttributes.RAW.name] = raw_upper[0]
-        self.air_trap_valve_states[TDAirTrapValves.H13_VALV.name] = h13_vlv_st[0]
-        self.air_trap_valve_states[TDAirTrapValves.H20_VALV.name] = h20_vlv_st[0]
+        self.air_trap_level_sensors[td_enum_repository.TDAirTrapLevelSensorNames.H17_LEVEL.name][td_enum_repository.TDAirTrapLevelSensorAttributes.LEVEL.name] = lower[0]
+        self.air_trap_level_sensors[td_enum_repository.TDAirTrapLevelSensorNames.H16_LEVEL.name][td_enum_repository.TDAirTrapLevelSensorAttributes.LEVEL.name] = upper[0]
+        self.air_trap_level_sensors[td_enum_repository.TDAirTrapLevelSensorNames.H17_LEVEL.name][td_enum_repository.TDAirTrapLevelSensorAttributes.RAW.name] = raw_lower[0]
+        self.air_trap_level_sensors[td_enum_repository.TDAirTrapLevelSensorNames.H16_LEVEL.name][td_enum_repository.TDAirTrapLevelSensorAttributes.RAW.name] = raw_upper[0]
+        self.air_trap_valve_states[td_enum_repository.TDAirTrapValveNames.H13_VALV.name] = h13_vlv_st[0]
+        self.air_trap_valve_states[td_enum_repository.TDAirTrapValveNames.H20_VALV.name] = h20_vlv_st[0]
         self.air_control = controlling[0]
         self.td_air_trap_timestamp = timestamp
 
@@ -219,8 +219,8 @@
             self.logger.debug("Timeout!!!!")
             return False
 
-    def cmd_set_td_air_trap_valve(self, valve: int = TDAirTrapValves.H13_VALV.value,
-                                  valve_state: int = TDAirTrapState.STATE_CLOSED.value) -> int:
+    def cmd_set_td_air_trap_valve(self, valve: int = td_enum_repository.TDAirTrapValveNames.H13_VALV.value,
+                                  valve_state: int = td_enum_repository.TDAirTrapState.STATE_CLOSED.value) -> int:
         """
         Constructs and sends an open/close command to the TD air trap valves.