diff --git a/docs/API/affine.rst b/docs/API/geometric.rst
similarity index 100%
rename from docs/API/affine.rst
rename to docs/API/geometric.rst
diff --git a/docs/API/image.rst b/docs/API/image.rst
index dd5d0bd..a445b59 100644
--- a/docs/API/image.rst
+++ b/docs/API/image.rst
@@ -8,6 +8,6 @@
    :maxdepth: 2
    :caption: API Documentation
 
-   affine
+   geometric
    augment
    filter
diff --git a/serket/nn/__init__.py b/serket/nn/__init__.py
index 0ccb679..d009faa 100644
--- a/serket/nn/__init__.py
+++ b/serket/nn/__init__.py
@@ -302,6 +302,4 @@
     "Upsample1D",
     "Upsample2D",
     "Upsample3D",
-    # block
-    "blocks",
 )