diff --git a/python/python/raphtory/__init__.pyi b/python/python/raphtory/__init__.pyi
index 0fce22b98..d2fa73825 100644
--- a/python/python/raphtory/__init__.pyi
+++ b/python/python/raphtory/__init__.pyi
@@ -1649,7 +1649,7 @@ class Graph:
         edge_label=None,
         colour_nodes_by_type=False,
         notebook=False,
-        **kwargs
+        **kwargs,
     ):
         """
         Draw a graph with PyVis.
@@ -4143,7 +4143,7 @@ class PersistentGraph:
         edge_label=None,
         colour_nodes_by_type=False,
         notebook=False,
-        **kwargs
+        **kwargs,
     ):
         """
         Draw a graph with PyVis.
diff --git a/raphtory/src/python/graph/graph_with_deletions.rs b/raphtory/src/python/graph/graph_with_deletions.rs
index c92a4c877..654350ce8 100644
--- a/raphtory/src/python/graph/graph_with_deletions.rs
+++ b/raphtory/src/python/graph/graph_with_deletions.rs
@@ -533,7 +533,15 @@ impl PyPersistentGraph {
         layer: Option<&str>,
         layer_col: Option<&str>,
     ) -> Result<(), GraphError> {
-        load_edge_deletions_from_pandas(&self.graph.0, df, time, src, dst, layer, layer_col)
+        load_edge_deletions_from_pandas(
+            self.graph.core_graph(),
+            df,
+            time,
+            src,
+            dst,
+            layer,
+            layer_col,
+        )
     }
 
     /// Load edges deletions from a Parquet file into the graph.