diff --git a/tests/test_functional.py b/tests/test_functional.py
index 697b01d..8ef6a3d 100644
--- a/tests/test_functional.py
+++ b/tests/test_functional.py
@@ -12,7 +12,7 @@
 _fakecmd = os.path.join(os.path.dirname(__file__), 'fakecmd.py')
 
 
-def setup():
+def setup_module():
     global _tempdir, _srcdir, _outdir
     _tempdir = tempfile.mkdtemp()
     _srcdir = os.path.join(_tempdir, 'src')
@@ -20,7 +20,7 @@ def setup():
     os.mkdir(_srcdir)
 
 
-def teardown():
+def teardown_module():
     shutil.rmtree(_tempdir)
 
 
diff --git a/tests/test_svgstyle.py b/tests/test_svgstyle.py
index c62e1ed..a25901f 100644
--- a/tests/test_svgstyle.py
+++ b/tests/test_svgstyle.py
@@ -5,12 +5,12 @@
 from sphinxcontrib import plantuml
 
 
-def setup():
+def setup_module():
     global _tempdir
     _tempdir = tempfile.mkdtemp()
 
 
-def teardown():
+def teardown_module():
     shutil.rmtree(_tempdir)