1 files changed, 8 insertions, 5 deletions

diff --git a/sway/commands/unmark.c b/sway/commands/unmark.c
index cedfcfb2..c3a6ac4b 100644
--- a/sway/commands/unmark.c
+++ b/sway/commands/unmark.c
@@ -8,9 +8,13 @@
 #include "log.h"
 #include "stringop.h"
 
-static void remove_all_marks_iterator(struct sway_container *con, void *data) {
+static void remove_mark(struct sway_container *con) {
         container_clear_marks(con);
-        container_update_marks_textures(con);
+        container_update_marks(con);
+}
+
+static void remove_all_marks_iterator(struct sway_container *con, void *data) {
+        remove_mark(con);
 }
 
 // unmark                  Remove all marks from all views
@@ -21,7 +25,7 @@ static void remove_all_marks_iterator(struct sway_container *con, void *data) {
 struct cmd_results *cmd_unmark(int argc, char **argv) {
         // Determine the container
         struct sway_container *con = NULL;
-        if (config->handler_context.using_criteria) {
+        if (config->handler_context.node_overridden) {
                 con = config->handler_context.container;
         }
 
@@ -38,8 +42,7 @@ struct cmd_results *cmd_unmark(int argc, char **argv) {
                 }
         } else if (con && !mark) {
                 // Clear all marks from the given container
-                container_clear_marks(con);
-                container_update_marks_textures(con);
+                remove_mark(con);
         } else if (!con && mark) {
                 // Remove mark from whichever container has it
                 container_find_and_unmark(mark);