diff --git a/dkist/dataset/tests/test_tiled_dataset.py b/dkist/dataset/tests/test_tiled_dataset.py
index f5eaecc8..fa477990 100644
--- a/dkist/dataset/tests/test_tiled_dataset.py
+++ b/dkist/dataset/tests/test_tiled_dataset.py
@@ -80,7 +80,8 @@ def test_tiled_dataset_from_components(dataset):
 @pytest.mark.remote_data
 @pytest.mark.parametrize("share_zscale", [True, False], ids=["share_zscale", "indpendent_zscale"])
 def test_tileddataset_plot(share_zscale):
-    from dkist.data.sample import VBI_AJQWW
+    from dkist.data.sample import VBI_AJQWW, download_all_sample_data
+    download_all_sample_data()
     ori_ds = load_dataset(VBI_AJQWW)
 
     newtiles = []
@@ -107,7 +108,8 @@ def test_tileddataset_plot(share_zscale):
 def test_tileddataset_plot_limit_swapping(swap_tile_limits):
     # Also test that row/column sizes are correct
 
-    from dkist.data.sample import VBI_AJQWW
+    from dkist.data.sample import VBI_AJQWW, download_all_sample_data
+    download_all_sample_data()
     ori_ds = load_dataset(VBI_AJQWW)
 
     # Swap WCS to make the `swap_tile_limits` option more natural
@@ -150,7 +152,8 @@ def test_tileddataset_plot_limit_swapping(swap_tile_limits):
 
 @pytest.mark.remote_data
 def test_tileddataset_plot_non2d_sliceindex():
-    from dkist.data.sample import VBI_AJQWW
+    from dkist.data.sample import VBI_AJQWW, download_all_sample_data
+    download_all_sample_data()
     ds = load_dataset(VBI_AJQWW)
 
     newtiles = []
diff --git a/tox.ini b/tox.ini
index cbc94054..4f39ade1 100644
--- a/tox.ini
+++ b/tox.ini
@@ -61,7 +61,6 @@ commands_pre =
     oldestdeps: minimum_dependencies dkist --filename requirements-min.txt
     # We need to pin down the cryptography transitive dependency because of globus
     oldestdeps: pip install -r requirements-min.txt cryptography<42
-    figure: python -c "from dkist.data.sample import download_all_sample_data; download_all_sample_data()"
     pip freeze --all --no-input
 commands =
     figure: /bin/sh -c "mkdir -p ./figure_test_images; python -c 'import matplotlib as mpl; print(mpl.ft2font.__file__, mpl.ft2font.__freetype_version__, mpl.ft2font.__freetype_build_type__)' > ./figure_test_images/figure_version_info.txt"