From 8862696926173104957729683832591438161557 Mon Sep 17 00:00:00 2001
From: Glenn Smith <glenn@vector35.com>
Date: Fri, 27 Dec 2024 16:14:46 -0500
Subject: Render Layers

---
 rust/src/medium_level_il/function.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'rust/src/medium_level_il/function.rs')

diff --git a/rust/src/medium_level_il/function.rs b/rust/src/medium_level_il/function.rs
index a9802005..8f91c731 100644
--- a/rust/src/medium_level_il/function.rs
+++ b/rust/src/medium_level_il/function.rs
@@ -548,10 +548,10 @@ impl MediumLevelILFunction {
         unsafe { BNGetMediumLevelILSSAVarValue(self.handle, &raw_var, ssa_variable.version) }.into()
     }
 
-    pub fn create_graph(&self, settings: Option<DisassemblySettings>) -> FlowGraph {
+    pub fn create_graph(&self, settings: Option<DisassemblySettings>) -> Ref<FlowGraph> {
         let settings = settings.map(|x| x.handle).unwrap_or(std::ptr::null_mut());
         let graph = unsafe { BNCreateMediumLevelILFunctionGraph(self.handle, settings) };
-        unsafe { FlowGraph::from_raw(graph) }
+        unsafe { FlowGraph::ref_from_raw(graph) }
     }
 
     /// This gets just the MLIL variables - you may be interested in the union
-- 
cgit v1.3.1