Fix some pretty-printing

1 files changed, 6 insertions, 6 deletions

diff --git a/src/AST/Pretty.hs b/src/AST/Pretty.hs
index c7b1497..5e01af8 100644
--- a/src/AST/Pretty.hs
+++ b/src/AST/Pretty.hs
@@ -139,13 +139,13 @@ ppExpr' d val expr = case expr of
     a' <- ppExpr' 11 val a
     name <- genNameIfUsedIn' "i" (tTup (sreplicate n tIx)) IZ b
     e' <- ppExpr' 0 (Const name `SCons` val) b
-    return $ group $ flatAlt
-      (ppParen (d > 0) $
-         hang 2 $
+    return $ ppParen (d > 0) $
+      group $ flatAlt
+        (hang 2 $
            annotate AHighlight (ppString "build") <> ppX expr <+> a'
            <+> ppString "$" <+> ppString "\\" <> ppString name <+> ppString "->"
            <> hardline <> e')
-      (ppApp (annotate AHighlight (ppString "build") <> ppX expr) [a', ppLam [ppString name] e'])
+        (ppApp (annotate AHighlight (ppString "build") <> ppX expr) [a', ppLam [ppString name] e'])
 
   EFold1Inner _ a b c -> do
     name1 <- genNameIfUsedIn (typeOf a) (IS IZ) a
@@ -251,8 +251,8 @@ ppExpr' d val expr = case expr of
     return $ ppParen (d > 10) $
       ppApp (annotate AMonoid (ppString "accum") <> ppX expr) [ppString (show (fromSNat i)), e1', e2', e3']
 
-  EZero _ t -> return $ parens $
-    annotate AMonoid (ppString "zero") <> ppX expr <+> ppString "::" <+> ppSTy' 0 t <> ppString ")"
+  EZero _ t -> return $ ppParen (d > 0) $
+    annotate AMonoid (ppString "zero") <> ppX expr <+> ppString "@" <> ppSTy' 11 t
 
   EPlus _ _ a b -> do
     a' <- ppExpr' 11 val a