1 files changed, 8 insertions, 2 deletions

diff --git a/src/Ewmh.cc b/src/Ewmh.cc
index 843b8d2..46bb94e 100644
--- a/src/Ewmh.cc
+++ b/src/Ewmh.cc
@@ -241,7 +241,8 @@ void Ewmh::setupFrame(FluxboxWindow &win) {
             win.setWorkspace(desktop);
 
         XFree(data);
-    }
+    } else 
+        updateWorkspace(win);
 
 }
 
@@ -265,6 +266,10 @@ void Ewmh::updateFocusedWindow(BScreen &screen, Window win) {
                                        (unsigned char *)&win, 1);
 }
 
+void Ewmh::updateClientClose(WinClient &winclient){
+    updateClientList(winclient.screen());
+}
+
 void Ewmh::updateClientList(BScreen &screen) {
     size_t num=0;
 
@@ -568,7 +573,8 @@ void Ewmh::updateHints(FluxboxWindow &win) {
 }
 
 void Ewmh::updateWorkspace(FluxboxWindow &win) {
-    long workspace = win.workspaceNumber();
+    long workspace = win.isInitialized() ? win.workspaceNumber() : win.screen().currentWorkspaceID();
+
     if (win.isStuck())
         workspace = 0xFFFFFFFF; // appear on all desktops/workspaces