diff --git a/src/main/python/docs/source/guide/federated.rst b/src/main/python/docs/source/guide/federated.rst
index fefea544945..1e6eec7f9b8 100644
--- a/src/main/python/docs/source/guide/federated.rst
+++ b/src/main/python/docs/source/guide/federated.rst
@@ -89,9 +89,9 @@ The print should look like
 
 .. code-block::
 
-  [[ 1.  4.  9.  1.  4.  9.]
-   [16. 25. 36. 16. 25. 36.]
-   [49. 64. 81. 49. 64. 81.]]
+  [[198. 243. 288.]
+   [198. 243. 288.]
+   [198. 243. 288.]]
 
 .. note::