1 files changed, 8 insertions, 8 deletions
diff --git a/src/gui/tasks/table/model.py b/src/gui/tasks/table/model.py
index 5013569..062ded6 100644
--- a/src/gui/tasks/table/model.py
+++ b/src/gui/tasks/table/model.py
@@ -139,17 +139,17 @@ def is_reversed(row: int, order: Qt.SortOrder) -> bool:
         return order == Qt.DescendingOrder
 
 def difficulty_color(d: Difficulty) -> QtGui.QColor:
-    if d == Difficulty.LOW:
-        return gui.color.low_difficulty
-    elif d == Difficulty.MEDIUM:
-        return gui.color.medium_difficulty
-    elif d == Difficulty.HIGH:
-        return gui.color.high_difficulty
+    if d == Difficulty.EASY:
+        return gui.color.easy_difficulty
+    elif d == Difficulty.NORMAL:
+        return gui.color.normal_difficulty
+    elif d == Difficulty.HARD:
+        return gui.color.hard_difficulty
 
 def priority_color(p: Priority) -> QtGui.QColor:
     if p == Priority.LOW:
         return gui.color.low_priority
-    elif p == Priority.MEDIUM:
-        return gui.color.medium_priority
+    elif p == Priority.MIDDLE:
+        return gui.color.middle_priority
     elif p == Priority.HIGH:
         return gui.color.high_priority