Fix ndcube issue

DKISTDC · Jan 22, 2025 · 0d0f247 · 0d0f247
1 parent 8156d8b
commit 0d0f247
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dkist/dataset/tests/test_tiled_dataset.py b/dkist/dataset/tests/test_tiled_dataset.py
@@ -85,7 +85,7 @@ def test_tileddataset_plot(share_zscale):
     # https://github.com/sunpy/ndcube/issues/815
     for tile in newtiles:
         tile.meta["inventory"] = ori_ds.inventory
-    ds = TiledDataset(np.array(newtiles).reshape(ds.shape), inventory=newtiles[0].inventory)
+    ds = TiledDataset(np.array(newtiles).reshape(ori_ds.shape), inventory=newtiles[0].inventory)
     fig = plt.figure(figsize=(12, 15))
     ds.plot(0, share_zscale=share_zscale, fig=fig)
     return plt.gcf()