1 files changed, 4 insertions, 6 deletions

diff --git a/dht/Presence/Nesting.hs b/dht/Presence/Nesting.hs
index 403d63cf..a9e550d4 100644
--- a/dht/Presence/Nesting.hs
+++ b/dht/Presence/Nesting.hs
@@ -24,12 +24,10 @@ doNestingXML :: Monad m => NestingXML o m r -> ConduitM Event o m r
 doNestingXML m =
     evalStateC (XMLState 0 StrictNil) (trackNesting .| m)
 
-startNestingXML :: Monad m => NestingXML o m r -> ConduitM Event o m (r, XMLState)
-startNestingXML m =
-    runStateC (XMLState 0 StrictNil) (trackNesting .| m)
-
-finishNestingXML :: Monad m => XMLState -> NestingXML o m r -> ConduitM Event o m r
-finishNestingXML = evalStateC
+startNestingXML :: Monad m => NestingXML o m r -> ConduitM Event o m (r, NestingXML o m ())
+startNestingXML m = do
+    (r,st) <- runStateC (XMLState 0 StrictNil) (trackNesting .| m)
+    return (r, lift $ put st)
 
 nesting :: Monad m => NestingXML o m Int
 nesting = lift $ (return . nestingLevel) =<< get