diff --git a/acme/types.py b/acme/types.py
index abfeebc072..7d841b97a5 100644
--- a/acme/types.py
+++ b/acme/types.py
@@ -23,11 +23,13 @@
 NestedArray = Any
 NestedTensor = Any
 
+# pytype: disable=not-supported-yet
 NestedSpec = Union[
     specs.Array,
     Iterable['NestedSpec'],
-    Mapping[Any, 'NestedSpec'],  # pytype: disable=not-supported-yet
+    Mapping[Any, 'NestedSpec'],
 ]
+# pytype: enable=not-supported-yet
 
 # TODO(b/144763593): Replace all instances of nest with the tensor/array types.
 Nest = Union[NestedArray, NestedTensor, NestedSpec]