Index: leahi_dialin/fp/modules/water_pumps.py
===================================================================
diff -u -r77740bc2ccace473a471515cadd4bc1353360cd6 -r9ac39d94327325e5b882318bc0e5126668eec66e
--- leahi_dialin/fp/modules/water_pumps.py	(.../water_pumps.py)	(revision 77740bc2ccace473a471515cadd4bc1353360cd6)
+++ leahi_dialin/fp/modules/water_pumps.py	(.../water_pumps.py)	(revision 9ac39d94327325e5b882318bc0e5126668eec66e)
@@ -16,7 +16,9 @@
 import struct
 from enum import unique
 from logging import Logger
+from enum import unique
 
+from leahi_dialin.utils.base import DialinEnum
 from .constants import RESET, NO_RESET
 from leahi_dialin.common.msg_defs import MsgIds, MsgFieldPositions
 from leahi_dialin.protocols.CAN import DenaliMessage, DenaliChannels