Fix alignment of top level braces in PseudoC

1 files changed, 3 insertions, 0 deletions

diff --git a/lang/c/pseudoc.cpp b/lang/c/pseudoc.cpp
index 2a1ac3a9..8de510a7 100644
--- a/lang/c/pseudoc.cpp
+++ b/lang/c/pseudoc.cpp
@@ -31,6 +31,7 @@ void PseudoCFunction::BeginLines(const HighLevelILInstruction& instr, HighLevelI
 	if (instr.exprIndex == m_highLevelIL->GetRootExpr().exprIndex)
 	{
 		// At top level, add braces around the entire function
+		tokens.PrependCollapseIndicator();
 		tokens.AppendOpenBrace();
 		tokens.NewLine();
 		tokens.IncreaseIndent();
@@ -45,6 +46,7 @@ void PseudoCFunction::EndLines(const HighLevelILInstruction& instr, HighLevelILT
 		// At top level, add braces around the entire function
 		tokens.NewLine();
 		tokens.DecreaseIndent();
+		tokens.PrependCollapseIndicator();
 		tokens.AppendCloseBrace();
 	}
 }
@@ -813,6 +815,7 @@ void PseudoCFunction::GetExprTextInternal(const HighLevelILInstruction& instr, H
 				if (function->IsInstructionCollapsed(instr, 1))
 				{
 					tokens.Append(CollapsedInformationToken, " {...}");
+					tokens.NewLine();
 				}
 				else
 				{