diff --git a/docs/src/python/nn/layers.rst b/docs/src/python/nn/layers.rst
index cb31d4be0..5ef45d60d 100644
--- a/docs/src/python/nn/layers.rst
+++ b/docs/src/python/nn/layers.rst
@@ -20,7 +20,7 @@ Layers
    Linear
    Conv1d
    Conv2d
-   BatchNorm1d
+   BatchNorm
    LayerNorm
    RMSNorm
    GroupNorm