diff --git a/torchdata/dataloader2/dataloader2.py b/torchdata/dataloader2/dataloader2.py
index 15b582187..1ef07b891 100644
--- a/torchdata/dataloader2/dataloader2.py
+++ b/torchdata/dataloader2/dataloader2.py
@@ -213,6 +213,7 @@ def __iter__(self) -> DataLoader2Iterator[T_co]:
                     self.datapipe = self.reading_service.restore(self.datapipe, self.reading_service_state)
                 self._adapted = True
 
+            # TODO: To allow arbitrary checkpointing, add and execute `restore_iteration` here instead.
             if self.reading_service is not None:
                 iter_reset_fn = self.reading_service.initialize_iteration(self._seed_generator)
                 if iter_reset_fn: