diff --git a/pyfstat/tcw_fstat_map_funcs.py b/pyfstat/tcw_fstat_map_funcs.py
index e3ef1f34c2e6e3a4149cc5a0a66c14403bfad2d3..10ee31f4cdaed84de2660134b978d3b8d7a0a76c 100644
--- a/pyfstat/tcw_fstat_map_funcs.py
+++ b/pyfstat/tcw_fstat_map_funcs.py
@@ -31,11 +31,8 @@ def _optional_import(modulename, shorthand=None):
         logging.debug("Successfully imported module %s%s." % (modulename, shorthandbit))
         success = True
     except ImportError as e:
-        if e.message == "No module named " + modulename:
-            logging.debug("No module {:s} found.".format(modulename))
-            success = False
-        else:
-            raise
+        logging.debug("Failed to import module {:s}.".format(modulename))
+        success = False
 
     return success
 
diff --git a/setup.py b/setup.py
index 7e8c8637bdaae3bb74b9eafae0db449007d51ec3..362ab3b1d2c99923f65f8a74c8034bfc876b1dce 100644
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,5 @@ setup(
         "bashplotlib",
         "peakutils",
         "pathos",
-        "pycuda",
     ],
 )
diff --git a/tests.py b/tests.py
index a9822733e84bf182ac65c0ab23f37375fd567b4a..26275f5bc03586e6c43571e2c71555cdda11533f 100644
--- a/tests.py
+++ b/tests.py
@@ -5,7 +5,7 @@ import shutil
 import pyfstat
 import lalpulsar
 import logging
-
+import time
 
 class Test(unittest.TestCase):
     outdir = "TestData"
@@ -85,6 +85,7 @@ class Writer(Test):
         Writer.run_makefakedata()
         time_second = os.path.getmtime(Writer.sftfilepath)
         self.assertTrue(time_first == time_second)
+        time.sleep(1)  # make sure timestamp is actually different!
         os.system("touch {}".format(Writer.config_file_name))
         Writer.run_makefakedata()
         time_third = os.path.getmtime(Writer.sftfilepath)
@@ -550,12 +551,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(