diff --git a/docs/customization.qmd b/docs/customization.qmd
index 265103a..78c21e0 100644
--- a/docs/customization.qmd
+++ b/docs/customization.qmd
@@ -207,13 +207,14 @@ Here's an entity-relationship diagram (ERD) of the current core database schema,
 import sys
 sys.path.append("..")
 from utils.core.models import *
-from utils.core.db import engine
+from utils.core.db import get_connection_url
+from sqlmodel import create_engine
 from sqlalchemy import MetaData
 from sqlalchemy_schemadisplay import create_schema_graph
 
 # Create the directed graph
 graph = create_schema_graph(
-    engine=engine,
+    engine=create_engine(get_connection_url()),
     metadata=SQLModel.metadata,
     show_datatypes=True,
     show_indexes=True,