diff --git a/dkist/dataset/tests/test_plotting.py b/dkist/dataset/tests/test_plotting.py
index 6586bf13..0b9a76cf 100644
--- a/dkist/dataset/tests/test_plotting.py
+++ b/dkist/dataset/tests/test_plotting.py
@@ -9,18 +9,22 @@
 def test_dataset_projection(dataset_3d, aslice):
     pytest.importorskip("ndcube", "2.0.2")  # https://github.com/sunpy/ndcube/pull/509
     ds = dataset_3d[aslice]
+    fig = plt.figure()
     ax = plt.subplot(projection=ds)
     assert isinstance(ax, WCSAxes)
+    return fig
 
 
 # @pytest.mark.mpl_image_compare
 @pytest.mark.parametrize("aslice", (np.s_[0, :, :], np.s_[:, 0, :], np.s_[:, :, 0]))
 def test_2d_plot(dataset_3d, aslice):
+    fig = plt.figure()
     dataset_3d[aslice].plot()
-    return plt.gcf()
+    return fig
 
 
 # @pytest.mark.mpl_image_compare
 def test_2d_plot2(dataset_3d):
+    fig = plt.figure()
     dataset_3d[:, :, 0].plot(axes_units=["Angstrom", "deg", "deg"])
-    return plt.gcf()
+    return fig