diff --git a/tests/test_plot.py b/tests/test_plot.py
index 21bb490..bb22ec5 100644
--- a/tests/test_plot.py
+++ b/tests/test_plot.py
@@ -17,7 +17,7 @@
 SMALL_DATA = [1, 0, 1, 1, 0]
 
 
-def setup():
+def setup_module():
     TEST_FIGURES.mkdir(exist_ok=True)