diff --git a/tests.py b/tests.py
index a9822733e84bf182ac65c0ab23f37375fd567b4a..f3a43a023ec305cca7272d9bf832850e0caef411 100644
--- a/tests.py
+++ b/tests.py
@@ -550,12 +550,8 @@ class GridSearch(Test):
             tref=self.tref,
             Lambda0=[30, 0, 0, 0],
         )
-        search.run()
-        self.assertTrue(
-            os.path.isfile(
-                "{}/{}_slice_projection.png".format(search.outdir, search.label)
-            )
-        )
+        fig, axes = search.run(save=False)
+        self.assertTrue(fig is not None)
 
     def test_glitch_grid_search(self):
         search = pyfstat.GridGlitchSearch(