@@ -527,13 +527,13 @@ repeatM = Unfold step return
527-- /Internal/
527-- /Internal/
529{-# INLINE iterateM #-}
529{-# INLINE iterateM #-}
530−iterateM :: Monad m => (a -> m a) -> Unfold m a a
530+iterateM :: Monad m => (a -> m a) -> Unfold m (m a) a
531−iterateM f = Unfold step f
531+iterateM f = Unfold step id
533 {-# INLINE_LATE step #-}
533 {-# INLINE_LATE step #-}
534 step x = do
534 step x = do
535 fx <- f x
535 fx <- f x
536− return $ Yield fx fx
536+ return $ Yield x fx
538-- | @fromIndicesM gen@ generates an infinite stream of values using @gen@
538-- | @fromIndicesM gen@ generates an infinite stream of values using @gen@
539-- starting from the seed.
539-- starting from the seed.