diff --git a/docs/notebooks/sax_01_sax.py b/docs/notebooks/sax_01_sax.py
index b83460d7..8e1194a6 100644
--- a/docs/notebooks/sax_01_sax.py
+++ b/docs/notebooks/sax_01_sax.py
@@ -979,9 +979,7 @@ def compact_mzi():
 
 # %%
 delta_length = 10
-mzi_component = gf.components.mzi_phase_shifter_top_heater_metal(
-    delta_length=delta_length
-)
+mzi_component = gf.components.mzi_phase_shifter(delta_length=delta_length)
 mzi_component.plot()
 
 
@@ -1036,9 +1034,7 @@ def phase_shifter_heater(
 }
 
 # %%
-mzi_component = gf.components.mzi_phase_shifter_top_heater_metal(
-    delta_length=delta_length
-)
+mzi_component = gf.components.mzi_phase_shifter(delta_length=delta_length)
 netlist = sax.netlist(mzi_component.get_netlist())
 
 # %%