diff --git a/dkist/tests/generate_aia_dataset.py b/dkist/tests/generate_aia_dataset.py
index 14a02727..10b5aa0b 100644
--- a/dkist/tests/generate_aia_dataset.py
+++ b/dkist/tests/generate_aia_dataset.py
@@ -147,6 +147,8 @@ def generate_dataset_gwcs(wcs, wave_coords, time_coords):
 tree = {
     'dataset': references_from_filenames(cube, relative_to=path),
     'wcs': gwcs_object,
+    'wave_coords': wave_coords,
+    'time_coords': time_coords,
 }
 
 print(tree)