diff --git a/enums.py b/enums.py
index 20e1f67..0b21b42 100644
--- a/enums.py
+++ b/enums.py
@@ -1,6 +1,6 @@
-import enum
+from enum import Enum
 
 
-class VmStatus(enum):
+class VmStatus(Enum):
     REQUESTED_NEW = "REQUESTED_NEW"
     SCHEDULED_DEPLOY = "SCHEDULED_DEPLOY"