diff --git a/metrics/confusion_matrix/confusion_matrix.py b/metrics/confusion_matrix/confusion_matrix.py
index 1d0c7797..37e0f78f 100644
--- a/metrics/confusion_matrix/confusion_matrix.py
+++ b/metrics/confusion_matrix/confusion_matrix.py
@@ -40,7 +40,9 @@
         >>> confusion_matrix_metric = evaluate.load("confusion_matrix")
         >>> results = confusion_matrix_metric.compute(references=[0, 1, 2, 0, 1, 2], predictions=[0, 1, 1, 2, 1, 0])
         >>> print(results)  # doctest: +NORMALIZE_WHITESPACE
-        {'confusion_matrix': array([[1, 0, 1], [0, 2, 0], [1, 1, 0]])}
+        {'confusion_matrix': array([[1, 0, 1],
+                            [0, 2, 0],
+                            [1, 1, 0]], dtype=int64)}
 """