diff --git a/src/XMonad/Operations.hs b/src/XMonad/Operations.hs
index 6e4262b..86a22cd 100644
--- a/src/XMonad/Operations.hs
+++ b/src/XMonad/Operations.hs
@@ -456,7 +456,8 @@ filterMessageWithNoRefresh p a = updateLayoutsBy $ \ wrk ->
       else pure Nothing
 
 -- | Update the layouts of some workspaces.
-updateLayoutsBy :: (WindowSpace -> X (Maybe (Layout Window))) -> X ()
+updateLayoutsBy :: MonadState XState m
+                => (WindowSpace -> m (Maybe (Layout Window))) -> m ()
 updateLayoutsBy f = runOnWorkspaces' $ \ wrk ->
     maybe wrk (\ l' -> wrk{ W.layout = l' }) <$> f wrk