diff --git a/docs/examples/Wideband_TOA_walkthrough.py b/docs/examples/Wideband_TOA_walkthrough.py
index d2337cee3..9a1bf37f0 100644
--- a/docs/examples/Wideband_TOA_walkthrough.py
+++ b/docs/examples/Wideband_TOA_walkthrough.py
@@ -155,7 +155,7 @@
 # #### Design Matrix are combined
 
 # %%
-d_matrix, labels, units = fitter.get_designmatrix()
+d_matrix, labels, units, units_d = fitter.model.full_designmatrix(toas)
 
 # %%
 print("Number of TOAs:", toas.ntoas)