diff --git a/concept-maker_gui.py b/concept-maker_gui.py
index f6f0096..74e03e8 100644
--- a/concept-maker_gui.py
+++ b/concept-maker_gui.py
@@ -34,6 +34,7 @@ import threading
 import traceback
 import hashlib
 from dataclasses import dataclass
+from enum import Enum
 from pathlib import Path
 from typing import List, Dict, Optional, Set, Any