diff --git a/tfjs-core/src/ops/diag.ts b/tfjs-core/src/ops/diag.ts
index 366e83ea17..64741ad2d8 100644
--- a/tfjs-core/src/ops/diag.ts
+++ b/tfjs-core/src/ops/diag.ts
@@ -38,7 +38,7 @@ import {op} from './operation';
  * tf.diag(x).print()
  * ```
  * ```js
- * const x = tf.tensor2d([1, 2, 3, 4, 5, 6, 6, 8], [4, 2])
+ * const x = tf.tensor2d([1, 2, 3, 4, 5, 6, 7, 8], [4, 2])
  *
  * tf.diag(x).print()
  * ```