top inside braces

anoma · May 17, 2024 · c463f61 · c463f61
1 parent 84f2264
commit c463f61
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/src/Juvix/Compiler/Concrete/Print/Base.hs b/src/Juvix/Compiler/Concrete/Print/Base.hs
@@ -271,12 +271,11 @@ ppIterator isTop Iterator {..} = do
       rngs = ppCode <$> _iteratorRanges
       is' = parens . hsepSemicolon <$> nonEmpty is
       rngs' = parens . hsepSemicolon <$> nonEmpty rngs
-      b = ppMaybeTopExpression isTop _iteratorBody
-      b'
-        | _iteratorBodyBraces = braces (oneLineOrNextNoIndent b)
-        | otherwise = line <> b
+      b
+        | _iteratorBodyBraces = braces (oneLineOrNextNoIndent (ppTopExpressionType _iteratorBody))
+        | otherwise = line <> ppMaybeTopExpression isTop _iteratorBody
   parensIf _iteratorParens $
-    hang (n <+?> is' <+?> rngs' <> b')
+    hang (n <+?> is' <+?> rngs' <> b)
 
 instance PrettyPrint S.AName where
   ppCode n = annotated (AnnKind (S.getNameKind n)) (noLoc (pretty (n ^. S.anameVerbatim)))