1 files changed, 10 insertions, 2 deletions
diff --git a/src/graph/GraphContext.cpp b/src/graph/GraphContext.cpp
index 3f311145bc..5f33ed3537 100644
--- a/src/graph/GraphContext.cpp
+++ b/src/graph/GraphContext.cpp
@@ -22,7 +22,9 @@
  * SOFTWARE.
  */
 #include "arm_compute/graph/GraphContext.h"
-#include <arm_compute/graph.h>
+
+#include "arm_compute/graph.h"
+#include "arm_compute/graph/Utils.h"
 
 namespace arm_compute
 {
@@ -33,6 +35,12 @@ GraphContext::GraphContext()
 {
 }
 
+GraphContext::~GraphContext()
+{
+    _memory_managers.clear();
+    release_default_graph_context(*this);
+}
+
 const GraphConfig &GraphContext::config() const
 {
     return _config;
@@ -82,4 +90,4 @@ void GraphContext::finalize()
     }
 }
 } // namespace graph
-} // namespace arm_compute
-\ No newline at end of file
+} // namespace arm_compute