summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--StackSet.hs2
1 files changed, 1 insertions, 1 deletions
diff --git a/StackSet.hs b/StackSet.hs
index b7419f1..7607fb3 100644
--- a/StackSet.hs
+++ b/StackSet.hs
@@ -111,7 +111,7 @@ index k w = fmap (uncurry (++)) $ M.lookup k (stacks w)
view :: (Integral i, Integral j) => i -> StackSet i j a -> StackSet i j a
view n w | M.member n (stacks w)
= if M.member n (ws2screen w) then w { current = n }
- else tweak (fromJust $ screen (current w) w)
+ else maybe w tweak (screen (current w) w)
| otherwise = w
where
tweak sc = w { screen2ws = M.insert sc n (screen2ws w)