diff --git a/src/example_code.py b/src/example_code.py
index cf2716e..b50a7d1 100644
--- a/src/example_code.py
+++ b/src/example_code.py
@@ -269,7 +269,7 @@ def plot_compare_angle(theta1, ts1, label1, theta2, ts2, label2, title):
     mod = DWBAModel()
     ts_dwba = mod.calculate_ts(m)
 
-    plot_compare_angle(m['theta'], ts, 'DWBA', m['theta'], bmt[name], 'Benchmark', name)
+    plot_compare_angle(m['theta'], ts_dwba, 'DWBA', m['theta'], bmt[name], 'Benchmark', name)
 
 ##########################################################
 # And the DWBA and SDWBA on a real shape