diff --git a/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py b/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py
index 2c15130..23f9af9 100644
--- a/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py
+++ b/src/AI/DecisionTrees/projectSpecificClasses/DistFromObject.py
@@ -7,6 +7,6 @@ class DistFromObject(Enum):
 
     """
 
-    LT_3 = 0,
-    GE_3_LT_8 = 1,
+    LT_3 = 0
+    GE_3_LT_8 = 1
     GE_8 = 2
diff --git a/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py b/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py
index 70600f5..1fa339c 100644
--- a/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py
+++ b/src/AI/DecisionTrees/projectSpecificClasses/PlayerStatsValue.py
@@ -2,7 +2,7 @@ from enum import Enum
 
 
 class PlayerStatsValue(Enum):
-    ZERO_TO_QUARTER = 0,
-    QUARTER_TO_HALF = 1,
-    HALF_TO_THREE_QUARTERS = 2,
+    ZERO_TO_QUARTER = 0
+    QUARTER_TO_HALF = 1
+    HALF_TO_THREE_QUARTERS = 2
     THREE_QUARTERS_TO_FULL = 3
diff --git a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py
index 6d813e3..d41ab79 100644
--- a/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py
+++ b/src/AI/DecisionTrees/projectSpecificClasses/SurvivalClassification.py
@@ -6,6 +6,6 @@ class SurvivalClassification(Enum):
     For example food classification means that player should go for food.
     """
 
-    FOOD = 0,
-    WATER = 1,
+    FOOD = 0
+    WATER = 1
     REST = 2