diff --git a/docs/source/examples/full_workflows/plot_public_transit_assignment.py b/docs/source/examples/full_workflows/plot_public_transit_assignment.py
index 9bcbd204a..1d2671fb3 100644
--- a/docs/source/examples/full_workflows/plot_public_transit_assignment.py
+++ b/docs/source/examples/full_workflows/plot_public_transit_assignment.py
@@ -76,7 +76,7 @@
 
 # %%
 graph.create_edges()
-graph.edges.drop(columns="geometry")
+graph.edges
 
 # %%
 # Connector project matching
@@ -103,7 +103,7 @@
 graph_db.vertices.drop(columns="geometry")
 
 # %%
-graph_db.edges.drop(columns="geometry")
+graph_db.edges
 
 # %%
 # Converting to a AequilibraE graph object