diff --git a/miplearn/collectors/basic.py b/miplearn/collectors/basic.py
index a5dac1b..4b65cf2 100644
--- a/miplearn/collectors/basic.py
+++ b/miplearn/collectors/basic.py
@@ -50,6 +50,7 @@ class BasicCollector:
                         return
 
                 with H5File(h5_filename, "w") as h5:
+                    h5.put_scalar("data_filename", data_filename)
                     streams: List[Any] = [StringIO()]
                     if verbose:
                         streams += [sys.stdout]