diff --git a/src/Term.hs b/src/Term.hs index 62d281d71..105daceb9 100644 --- a/src/Term.hs +++ b/src/Term.hs @@ -39,6 +39,5 @@ termSize = cata size where alignCofreeWith :: Functor f => (forall a b. f a -> f b -> Maybe (f (These a b))) -> (a -> b -> c) -> These (Cofree f a) (Cofree f b) -> Free (CofreeF f c) (These (Cofree f a) (Cofree f b)) alignCofreeWith contrast combine = go where go terms = fromMaybe (pure terms) $ case terms of - These t1 t2 -> let (a1 :< s1, a2 :< s2) = (runCofree t1, runCofree t2) in - wrap . (combine a1 a2 :<) . fmap go <$> contrast s1 s2 + These t1 t2 -> wrap . (combine (extract t1) (extract t2) :<) . fmap go <$> contrast (unwrap t1) (unwrap t2) _ -> Nothing