diff --git a/dkist/dataset/tests/test_dataset.py b/dkist/dataset/tests/test_dataset.py
index 9b626385..aa8c7200 100644
--- a/dkist/dataset/tests/test_dataset.py
+++ b/dkist/dataset/tests/test_dataset.py
@@ -52,8 +52,7 @@ def test_repr(dataset, dataset_3d):
 
 
 def test_wcs_roundtrip(dataset):
-    # TOO generalise this so mark.accept_cli_dataset will work, if possible
-    p = (10*u.pixel, 10*u.pixel)
+    p = [1*u.pixel] * dataset.wcs.pixel_n_dim
     w = dataset.wcs.pixel_to_world(*p)
     p2 = dataset.wcs.world_to_pixel(w)
     assert_quantity_allclose(p, p2 * u.pix)