@@ -527,13 +527,13 @@ repeatM = Unfold step return
527527-- /Internal/
528528--
529529{-# INLINE iterateM #-}
530−iterateM :: Monad m => (a -> m a) -> Unfold m a a
531−iterateM f = Unfold step f
530+iterateM :: Monad m => (a -> m a) -> Unfold m (m a) a
531+iterateM f = Unfold step id
532532 where
533533 {-# INLINE_LATE step #-}
534534 step x = do
535535 fx <- f x
536− return $ Yield fx fx
536+ return $ Yield x fx
537537
538538-- | @fromIndicesM gen@ generates an infinite stream of values using @gen@
539539-- starting from the seed.