diff --git a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
index f1653cf4e660c7202e1198b083d8072dce592553..1eaaaeb284f15458de69e6151bf5c001e3e685e8 100644
--- a/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
+++ b/notebooks/REMI/REMI_Digitize_and_Transform.ipynb
@@ -87,7 +87,7 @@
     "\n",
     "import pasha as psh\n",
     "from euxfel_bunch_pattern import indices_at_sase, indices_at_laser\n",
-    "from extra_data import RunDirectory\n",
+    "from extra_data import RunDirectory, by_id\n",
     "from extra_remi import Analysis, trigger_dt\n",
     "from extra_remi.util import timing\n",
     "from extra_remi.rd_resort import signal_dt, hit_dt\n",
@@ -1152,7 +1152,10 @@
     "    with DataFile.from_details(out_folder, out_aggregator, run, seq_id) as outp:\n",
     "        outp.create_metadata(like=dc, proposal=proposal, run=run, sequence=seq_id,\n",
     "                             control_sources=control_sources, instrument_channels=instrument_channels)\n",
-    "        outp.create_index(seq_train_ids)\n",
+    "        outp.create_index(\n",
+    "            seq_train_ids, \n",
+    "            timestamps=dc.select_trains(by_id[seq_train_ids]).train_timestamps().astype(np.uint64)\n",
+    "        )\n",
     "        \n",
     "        for det_name in remi['detector']:\n",
     "            cur_device_id = det_device_id.format(karabo_id=karabo_id, det_name=det_name.upper())\n",