From f5de4bbfec37956d62138b66b57d54ddf18a2f77 Mon Sep 17 00:00:00 2001
From: Nils Vreman <nils.vreman@gmail.com>
Date: Fri, 9 Nov 2018 13:03:20 +0100
Subject: [PATCH] Only markUpdate left to fix

---
 gui/tasksetgui/TasksetDisplay.java | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/gui/tasksetgui/TasksetDisplay.java b/gui/tasksetgui/TasksetDisplay.java
index 20663d7..ef54185 100644
--- a/gui/tasksetgui/TasksetDisplay.java
+++ b/gui/tasksetgui/TasksetDisplay.java
@@ -99,7 +99,6 @@ public class TasksetDisplay extends BoxPanel implements Observer, UpdateListener
         marked = Optional.of((TaskGrid) obj);
 
         for (TaskGrid tg : gridList) {
-            //if (marked.map(taskgrid -> taskgrid.equals(tg)).orElse(false)) marked = Optional.of(tg);
             tg.unmark();
         }
 
@@ -144,6 +143,10 @@ public class TasksetDisplay extends BoxPanel implements Observer, UpdateListener
             if (prio != -1) {
                 taskset.removeTask(prio);
                 marked = Optional.empty();
+                // Unmarks the rest of the grid
+                for (TaskGrid tg : gridList) {
+                    tg.unmark();
+                }
             }
         } catch (NoSuchElementException e) {}
     }
-- 
GitLab