diff --git a/tfjs-core/src/ops/batchnorm2d.ts b/tfjs-core/src/ops/batchnorm2d.ts
index 08e295a2053..a53f78e1514 100644
--- a/tfjs-core/src/ops/batchnorm2d.ts
+++ b/tfjs-core/src/ops/batchnorm2d.ts
@@ -51,7 +51,7 @@ function batchNorm2d_(
   }
   util.assert(
       $x.rank === 2,
-      () => `Error in batchNorm3D: x must be rank 3 but got rank ` +
+      () => `Error in batchNorm2D: x must be rank 2 but got rank ` +
           `${$x.rank}.`);
   util.assert(
       $mean.rank === 2 || $mean.rank === 1,