Fix segfault when reaping invisible workspaces

2 files changed, 4 insertions, 2 deletions

diff --git a/sway/tree/container.c b/sway/tree/container.c
index 778108b4..c3cf6c64 100644
--- a/sway/tree/container.c
+++ b/sway/tree/container.c
@@ -79,7 +79,7 @@ struct sway_container *container_destroy(struct sway_container *cont) {
                 list_free(cont->marks);
         }
         if (parent) {
-                container_remove_child(cont);
+                parent = container_remove_child(cont);
         }
         if (cont->name) {
                 free(cont->name);
diff --git a/sway/tree/layout.c b/sway/tree/layout.c
index 32e6a77c..588ceb2d 100644
--- a/sway/tree/layout.c
+++ b/sway/tree/layout.c
@@ -110,9 +110,11 @@ struct sway_container *container_reap_empty(struct sway_container *container) {
         while (container->children->length == 0) {
                 if (container->type == C_WORKSPACE) {
                         if (!workspace_is_visible(container)) {
+                                struct sway_container *parent = container->parent;
                                 container_workspace_destroy(container);
+                                return parent;
                         }
-                        break;
+                        return container;
                 } else if (container->type == C_CONTAINER) {
                         struct sway_container *parent = container->parent;
                         container_destroy(container);