correct Moore diagonalization

ekmett · Mar 2, 2014 · d5b640f · d5b640f
1 parent b692fbe
commit d5b640f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Data/Machine/Moore.hs b/src/Data/Machine/Moore.hs
@@ -90,8 +90,8 @@ instance Pointed (Moore a) where
 instance Monad (Moore a) where
   return a = r where r = Moore a (const r)
   {-# INLINE return #-}
-  Moore a k >>= f = case f a of
-    Moore b _ -> Moore b (k >=> f)
+  k >>= f = j (fmap f k) where
+    j (Moore a g) = Moore (extract a) (\x -> j $ fmap (\(Moore _ h) -> h x) (g x))
   _ >> m = m
 
 instance Copointed (Moore a) where