diff --git a/src/sage/tests/books/computational-mathematics-with-sagemath/graphique_doctest.py b/src/sage/tests/books/computational-mathematics-with-sagemath/graphique_doctest.py
index aa153fd4cd..eebbe87aff 100644
--- a/src/sage/tests/books/computational-mathematics-with-sagemath/graphique_doctest.py
+++ b/src/sage/tests/books/computational-mathematics-with-sagemath/graphique_doctest.py
@@ -134,11 +134,11 @@ Sage example in ./graphique.tex, line 1120::
   sage: t = srange(0, 5, 0.1); p = Graphics()
   sage: for k in srange(0, 10, 0.15):
   ....:       y = integrate.odeint(f, k, t)
-  ....:       p += line(zip(t, flatten(y)))
+  ....:       p += line(zip(t, y.flatten()))
   sage: t = srange(0, -5, -0.1); q = Graphics()
   sage: for k in srange(0, 10, 0.15):
   ....:       y = integrate.odeint(f, k, t)
-  ....:       q += line(zip(t, flatten(y)))
+  ....:       q += line(zip(t, y.flatten()))
   sage: y = var('y')
   sage: v = plot_vector_field((1, -cos(x*y)), (x,-5,5), (y,-2,11))
   sage: g = p + q + v; g.show()