refactor: move proc registration to after proc compilation

0xPolygonMiden · Jul 23, 2024 · a850c9b · a850c9b
1 parent d11dcca
commit a850c9b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/assembly/src/assembler/mod.rs b/assembly/src/assembler/mod.rs
@@ -387,6 +387,9 @@ impl Assembler {
             // Compile this procedure
             let procedure = self.compile_procedure(pctx, mast_forest_builder)?;
 
+            // register the procedure in the MAST forest
+            mast_forest_builder.make_root(procedure.body_node_id());
+
             // Cache the compiled procedure, unless it's the program entrypoint
             if is_entry {
                 compiled_entrypoint = Some(Arc::from(procedure));
@@ -432,8 +435,6 @@ impl Assembler {
             self.compile_body(proc.iter(), &mut proc_ctx, None, mast_forest_builder)?
         };
 
-        mast_forest_builder.make_root(proc_body_root);
-
         Ok(proc_ctx.into_procedure(proc_body_root))
     }