diff --git a/src/caked/hdf5.py b/src/caked/hdf5.py
index 22e5d78..f4c395e 100644
--- a/src/caked/hdf5.py
+++ b/src/caked/hdf5.py
@@ -69,8 +69,7 @@ def __getitem__(self, key: str):
 
     def __iter__(self):
         with h5py.File(self.save_path, "r") as f:
-            for key in f:
-                yield key
+            yield from f
 
     def get(self, key: str, default=None, to_torch: bool = False):
         try: