Fix opening/closing braces in PseudoRust

1 files changed, 2 insertions, 0 deletions

diff --git a/lang/rust/pseudorust.cpp b/lang/rust/pseudorust.cpp
index d12b8e32..265d3688 100644
--- a/lang/rust/pseudorust.cpp
+++ b/lang/rust/pseudorust.cpp
@@ -37,6 +37,7 @@ void PseudoRustFunction::BeginLines(const HighLevelILInstruction& instr, HighLev
 	if (instr.exprIndex == m_highLevelIL->GetRootExpr().exprIndex)
 	{
 		// At top level, add braces around the entire function
+		tokens.PrependCollapseIndicator();
 		tokens.AppendOpenBrace();
 		tokens.NewLine();
 		tokens.IncreaseIndent();
@@ -51,6 +52,7 @@ void PseudoRustFunction::EndLines(const HighLevelILInstruction& instr, HighLevel
 		// At top level, add braces around the entire function
 		tokens.NewLine();
 		tokens.DecreaseIndent();
+		tokens.PrependCollapseIndicator();
 		tokens.AppendCloseBrace();
 	}
 }