diff --git a/ai_commands.py b/ai_commands.py
index 3b78d44..f459070 100644
--- a/ai_commands.py
+++ b/ai_commands.py
@@ -2,7 +2,7 @@
 import logging
 import re
 import sys
-import Enum
+from enum import Enum
 from datetime import datetime
 from typing import Optional