From c5c93fc82b8929d04f62d241ca50228de60fa5f4 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty@vector35.com>
Date: Fri, 13 Jul 2018 18:43:34 -0400
Subject: Add ability to update custom flow graphs

---
 function.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'function.cpp')

diff --git a/function.cpp b/function.cpp
index 62a8e1b0..202f6d77 100644
--- a/function.cpp
+++ b/function.cpp
@@ -811,7 +811,7 @@ void Function::ApplyAutoDiscoveredType(Type* type)
 Ref<FlowGraph> Function::CreateFunctionGraph(BNFunctionGraphType type, DisassemblySettings* settings)
 {
 	BNFlowGraph* graph = BNCreateFunctionGraph(m_object, type, settings ? settings->GetObject() : nullptr);
-	return new FlowGraph(graph);
+	return new CoreFlowGraph(graph);
 }
 
 
@@ -1396,7 +1396,7 @@ Ref<FlowGraph> Function::GetUnresolvedStackAdjustmentGraph()
 	BNFlowGraph* graph = BNGetUnresolvedStackAdjustmentGraph(m_object);
 	if (!graph)
 		return nullptr;
-	return new FlowGraph(graph);
+	return new CoreFlowGraph(graph);
 }
 
 
-- 
cgit v1.3.1