diff --git a/tensorflow_datasets/core/dataset_info.py b/tensorflow_datasets/core/dataset_info.py
index 5d47d85dfb5..af928d61a2d 100644
--- a/tensorflow_datasets/core/dataset_info.py
+++ b/tensorflow_datasets/core/dataset_info.py
@@ -787,7 +787,7 @@ def read_from_directory(self, dataset_info_dir: epath.PathLike) -> None:
       # Otherwise, we restore the dataset_info.json value
       if field.type == field.TYPE_MESSAGE:
         field_value.MergeFrom(field_value_restored)
-      elif field.label == field.LABEL_REPEATED:
+      elif getattr(field, "is_repeated", getattr(field, "label", None) == getattr(field, "LABEL_REPEATED", 3)):
         del field_value[:]
         field_value.extend(field_value_restored)
       else: