diff --git a/docs/api_reference/checkpoint.v1.handler_registration.rst b/docs/api_reference/checkpoint.v1.handler_registration.rst
index 2a27d041a5..54abe97ee2 100644
--- a/docs/api_reference/checkpoint.v1.handler_registration.rst
+++ b/docs/api_reference/checkpoint.v1.handler_registration.rst
@@ -8,8 +8,7 @@ CheckpointableHandler Registration
 
 CheckpointableHandlerRegistry
 ------------------------------------------------------------
-.. autoclass:: registration.CheckpointableHandlerRegistry
-  :members:
+.. autoclass:: CheckpointableHandlerRegistry
 
 Utilities
 ------------------------------------------------------------------
diff --git a/docs/api_reference/checkpoint.v1.handlers.rst b/docs/api_reference/checkpoint.v1.handlers.rst
index a7bcd52103..68c1b8e6cd 100644
--- a/docs/api_reference/checkpoint.v1.handlers.rst
+++ b/docs/api_reference/checkpoint.v1.handlers.rst
@@ -18,10 +18,3 @@ Handlers
 .. autoclass:: PyTreeHandler
 .. autoclass:: ProtoHandler
 .. autoclass:: JsonHandler
-
-Registration
-------------------------------------------------------------
-.. autoclass:: CheckpointableHandlerRegistry
-.. autofunction:: global_registry
-.. autofunction:: local_registry
-.. autofunction:: register_handler
\ No newline at end of file