2 files changed, 19 insertions, 12 deletions

diff --git a/sway/tree/container.c b/sway/tree/container.c
index c4d21f0a..89a47151 100644
--- a/sway/tree/container.c
+++ b/sway/tree/container.c
@@ -1378,10 +1378,12 @@ void container_replace(struct sway_container *container,
                 struct sway_container *replacement) {
         enum sway_fullscreen_mode fullscreen = container->fullscreen_mode;
         bool scratchpad = container->scratchpad;
+        struct sway_workspace *ws = NULL;
         if (fullscreen != FULLSCREEN_NONE) {
                 container_fullscreen_disable(container);
         }
         if (scratchpad) {
+                ws = container->workspace;
                 root_scratchpad_show(container);
                 root_scratchpad_remove_container(container);
         }
@@ -1390,7 +1392,7 @@ void container_replace(struct sway_container *container,
                 container_detach(container);
         }
         if (scratchpad) {
-                root_scratchpad_add_container(replacement);
+                root_scratchpad_add_container(replacement, ws);
         }
         switch (fullscreen) {
         case FULLSCREEN_WORKSPACE:
diff --git a/sway/tree/root.c b/sway/tree/root.c
index 1dabc287..55d78487 100644
--- a/sway/tree/root.c
+++ b/sway/tree/root.c
@@ -54,7 +54,7 @@ void root_destroy(struct sway_root *root) {
         free(root);
 }
 
-void root_scratchpad_add_container(struct sway_container *con) {
+void root_scratchpad_add_container(struct sway_container *con, struct sway_workspace *ws) {
         if (!sway_assert(!con->scratchpad, "Container is already in scratchpad")) {
                 return;
         }
@@ -77,18 +77,23 @@ void root_scratchpad_add_container(struct sway_container *con) {
         container_detach(con);
         con->scratchpad = true;
         list_add(root->scratchpad, con);
-
-        struct sway_seat *seat = input_manager_current_seat();
-        struct sway_node *new_focus = NULL;
-        if (parent) {
-                arrange_container(parent);
-                new_focus = seat_get_focus_inactive(seat, &parent->node);
+        if (ws) {
+                workspace_add_floating(ws, con);
         }
-        if (!new_focus) {
-                arrange_workspace(workspace);
-                new_focus = seat_get_focus_inactive(seat, &workspace->node);
+
+        if (!ws) {
+                struct sway_seat *seat = input_manager_current_seat();
+                struct sway_node *new_focus = NULL;
+                if (parent) {
+                        arrange_container(parent);
+                        new_focus = seat_get_focus_inactive(seat, &parent->node);
+                }
+                if (!new_focus) {
+                        arrange_workspace(workspace);
+                        new_focus = seat_get_focus_inactive(seat, &workspace->node);
+                }
+                seat_set_focus(seat, new_focus);
         }
-        seat_set_focus(seat, new_focus);
 
         ipc_event_window(con, "move");
 }