2 files changed, 12 insertions, 4 deletions
diff --git a/Main.hs b/Main.hs
index 414a283..a149d34 100644
--- a/Main.hs
+++ b/Main.hs
@@ -52,9 +52,16 @@ main = do
 
     let initialWinset = new defaultLayout workspaces $ zipWith SD xinesc gaps
 
-        winset | ("--resume" : s : _) <- args
-               , [(x, "")]            <- reads s = W.ensureTags defaultLayout workspaces x
-               | otherwise                       = initialWinset
+        maybeRead s = case reads s of
+                            [(x, "")] -> Just x
+                            _         -> Nothing
+
+        winset = fromMaybe initialWinset $ do
+                    ("--resume" : s : _) <- return args
+                    ws                   <- maybeRead s
+                    return . W.ensureTags defaultLayout workspaces
+                           $ W.mapLayout (fromMaybe defaultLayout . maybeRead) ws
+
         gaps = take (length xinesc) $ defaultGaps ++ repeat (0,0,0,0)
 
         cf = XConf
diff --git a/XMonad.hs b/XMonad.hs
index 818586f..48f49ca 100644
--- a/XMonad.hs
+++ b/XMonad.hs
@@ -218,8 +218,9 @@ spawn x = io $ do
 restart :: Maybe String -> Bool -> X ()
 restart mprog resume = do
     prog <- maybe (io getProgName) return mprog
-    args <- if resume then gets (("--resume":) . return . show . windowset) else return []
+    args <- if resume then gets (("--resume":) . return . showWs . windowset) else return []
     catchIO (executeFile prog True args Nothing)
+ where showWs = show . mapLayout show
 
 -- | Run a side effecting action with the current workspace. Like 'when' but
 whenJust :: Maybe a -> (a -> X ()) -> X ()