diff --git a/controls/device.py b/controls/device.py
index 974cbc7..8e63d50 100644
--- a/controls/device.py
+++ b/controls/device.py
@@ -1,6 +1,7 @@
 from dataclasses import dataclass
 from typing import Optional, Collection, Any
-from abc import abstractmethod
+from abc import ABC, abstractmethod
+from enum import Enum, auto
 
 class DeviceLifecycleState(Enum):
     INIT = "init"