diff --git a/src/experiments/algorithms/colour_FE.py b/src/experiments/algorithms/colour_FE.py
index d385282..a7cb763 100644
--- a/src/experiments/algorithms/colour_FE.py
+++ b/src/experiments/algorithms/colour_FE.py
@@ -5,6 +5,7 @@ import os
 import numpy as np
 import csv
 from ...helpers.treenum import Tree
+from enum import Enum
 
 CSV_PATH = "src\\experiments\\algorithms\\data\\"
 IMG_PATH = "dataset\\accasia_1210048 (2023_09_28 12_19_26 UTC).JPG"