diff --git a/tests/controls/__init__.py b/tests/controls/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/controls/test_device.py b/tests/controls/test_device.py
index 51249ad..f99a2c1 100644
--- a/tests/controls/test_device.py
+++ b/tests/controls/test_device.py
@@ -9,4 +9,4 @@ class DeviceLifecycleStateTest(TestCase):
         pass
 
     def test_enum(self):
-        self.assertEqual(DeviceLifecycleStateTest["INIT"], DeviceLifecycleStateTest.INIT)
+        self.assertEqual(DeviceLifecycleState["INIT"], DeviceLifecycleState.INIT)
diff --git a/tests/equipment/__init__.py b/tests/equipment/__init__.py
new file mode 100644
index 0000000..e69de29