diff --git a/doc/source/code/figure/kernel_polynomial_method.py.diff b/doc/source/code/figure/kernel_polynomial_method.py.diff
index 754b56ba9dbf4da6e0648b3139d1b18ebab334c4..3ac532512ea1d3add0d5604c897aa0e9ceb49399 100644
--- a/doc/source/code/figure/kernel_polynomial_method.py.diff
+++ b/doc/source/code/figure/kernel_polynomial_method.py.diff
@@ -316,7 +316,7 @@
      # construct the Haldane model
      lat, fsyst = make_syst_topo()
      # find 'A' and 'B' sites in the unit cell at the center of the disk
-     where = lambda s: np.linalg.norm(s.pos) < 3
+     where = lambda s: np.linalg.norm(s.pos) < 1
  
      # component 'xx'
      s_factory = kwant.kpm.LocalVectors(fsyst, where)