diff --git a/src/iblphotometry_tests/test_plots.py b/src/iblphotometry_tests/test_plots.py
index 4f165f3..6ec9744 100644
--- a/src/iblphotometry_tests/test_plots.py
+++ b/src/iblphotometry_tests/test_plots.py
@@ -1,5 +1,4 @@
-# import sys
-# import unittest
+import unittest
 import pandas as pd
 from pathlib import Path
 
@@ -99,11 +98,11 @@ def test_class_plotsignalresponse(self):
         trials = pd.read_parquet(self.paths['trials_table_kcenia_pqt'])
         plotobj = plots.PlotSignalResponse()
         plotobj.set_data(trials, processed_signal, times)
-        fig, axs = plotobj.set_fig_layout()
-        plotobj.plot_trialsort_psth(fig, axs)
-        fig, ax = plt.subplots(1, 1)
+        _, axs = plotobj.set_fig_layout()
+        plotobj.plot_trialsort_psth(axs)
+        _, ax = plt.subplots(1, 1)
         plotobj.plot_processed_trialtick(ax)
-        plt.show()
+        # plt.show()
         plt.close('all')
 
     """
@@ -210,8 +209,9 @@ def test_plot_event_tick(self):
     #     app.exec_()
 
 
-# if __name__ == '__main__':
-#     suite = unittest.TestSuite()
-#     suite.addTest(TestPlotters("test_gui"))
-#     runner = unittest.TextTestRunner()
-#     runner.run(suite)
+if __name__ == '__main__':
+    unittest.main()
+    # suite = unittest.TestSuite()
+    # suite.addTest(TestPlotters())
+    # runner = unittest.TextTestRunner()
+    # runner.run(suite)