diff --git a/_doc/index.rst b/_doc/index.rst
index 9bdc4e2..e3cf09f 100644
--- a/_doc/index.rst
+++ b/_doc/index.rst
@@ -182,7 +182,7 @@ to know onnx for that. See :ref:`l-numpy-api-onnx`.
     x = np.array([[0.1, 0.2], [0.3, 0.4]], dtype=np.float32)
     y = np.array([[0.11, 0.22], [0.33, 0.44]], dtype=np.float32)
     res = jitted_myloss(x, y)
-    print(to_dot(jitted_myloss.get_onnx()))
+    print("DOT-SECTION", to_dot(jitted_myloss.get_onnx()))
 
 Older versions
 ++++++++++++++
diff --git a/onnx_array_api/plotting/dot_plot.py b/onnx_array_api/plotting/dot_plot.py
index af8ad22..fdcb893 100644
--- a/onnx_array_api/plotting/dot_plot.py
+++ b/onnx_array_api/plotting/dot_plot.py
@@ -110,7 +110,7 @@ def myloss(x, y):
         x = np.array([[0.1, 0.2], [0.3, 0.4]], dtype=np.float32)
         y = np.array([[0.11, 0.22], [0.33, 0.44]], dtype=np.float32)
         res = jitted_myloss(x, y)
-        print(to_dot(jitted_myloss.get_onnx()))
+        print("DOT-SECTION", to_dot(jitted_myloss.get_onnx()))
     """
     clean_label_reg1 = re.compile("\\\\x\\{[0-9A-F]{1,6}\\}")
     clean_label_reg2 = re.compile("\\\\p\\{[0-9P]{1,6}\\}")