diff --git a/src/TypedSession/State/Pipeline.hs b/src/TypedSession/State/Pipeline.hs
--- a/src/TypedSession/State/Pipeline.hs
+++ b/src/TypedSession/State/Pipeline.hs
@@ -228,7 +228,7 @@
         Nothing -> modify (IntMap.insert from' ft)
         Just ft1 -> when (ft1 /= ft) (throwError @(ProtocolError r bst) AStateOnlyBeUsedForTheSamePair)
   , \_ -> pure ()
-  , \_ -> pure id
+  , \_ -> pure (restoreWrapper @(IntMap (r, r)))
   , \_ -> pure ()
   , \_ -> pure ()
   , \_ -> pure ()
diff --git a/typed-session-state-algorithm.cabal b/typed-session-state-algorithm.cabal
--- a/typed-session-state-algorithm.cabal
+++ b/typed-session-state-algorithm.cabal
@@ -20,7 +20,7 @@
 -- PVP summary:     +-+------- breaking API changes
 --                  | | +----- non-breaking API additions
 --                  | | | +--- code changes with no API change
-version:            0.4.2.2
+version:            0.4.2.3
 
 -- A short (one-line) description of the package.
 synopsis: Automatically generate status for typed-session.