fix type mistake in CodeTypes.hs

diff --git a/GHC/HetMet/CodeTypes.hs b/GHC/HetMet/CodeTypes.hs
index 255e8b4..d629119 100644 (file)
--- a/GHC/HetMet/CodeTypes.hs
+++ b/GHC/HetMet/CodeTypes.hs
@@ -33,9 +33,9 @@ hetmet_flatten ::
    forall g .
     GArrowSTKC g =>
       forall x y.
-         <[ y ]>@g
+         <[ x -> y ]>@g
          ->
-         (g (GArrowUnit g) y)
+         (g x y)
 hetmet_flatten x = unG (pga_flatten x)
 
 -- After the flattening pass the argument and result types of this