diff --git a/CHANGELOG.md b/CHANGELOG.md
index aa3c8aae8..af7b78043 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -8,7 +8,8 @@ changes. Best viewed [here](https://google-grain.readthedocs.io/en/latest/change
 * New features:
   * Adds automated batching into shared memory when multiprocess prefetch is
     used to save one data copy.
-  * Adds experimental `ShapeDtypeStruct` to represent dataset element specs.
+  * Adds experimental `ShapeDtypeStructProtocol` and `ShapeDtypeStruct` to
+    represent dataset element specs.
 
 * Breaking changes:
 
diff --git a/grain/experimental.py b/grain/experimental.py
index efcd81d68..217719545 100644
--- a/grain/experimental.py
+++ b/grain/experimental.py
@@ -28,6 +28,7 @@
     DatasetOptions,
     ExecutionTrackingMode,
     ShapeDtypeStruct,
+    ShapeDtypeStructProtocol,
 )
 from grain._src.python.dataset.dataset import (
     apply_transformations,