diff --git a/src/pyhpp/manipulation/graph.cc b/src/pyhpp/manipulation/graph.cc
index 427575d..bb682de 100644
--- a/src/pyhpp/manipulation/graph.cc
+++ b/src/pyhpp/manipulation/graph.cc
@@ -1249,7 +1249,7 @@ void exposeGraph() {
            DocClassMethod(pathValidation));
 
   // DocClass(Graph)
-  class_<PyWGraph>(
+  class_<PyWGraph, PyWGraphPtr_t>(
       "Graph",
       init<const std::string&, const PyWDevicePtr_t&, const PyWProblemPtr_t&>())
 
diff --git a/src/pyhpp/manipulation/graph.hh b/src/pyhpp/manipulation/graph.hh
index a01cada..6840a03 100644
--- a/src/pyhpp/manipulation/graph.hh
+++ b/src/pyhpp/manipulation/graph.hh
@@ -227,6 +227,7 @@ struct PyWGraph {
   // Initialization
   void initialize();
 };
+typedef std::shared_ptr<PyWGraph> PyWGraphPtr_t;
 
 }  // namespace manipulation
 }  // namespace pyhpp