diff --git a/tests/base.py b/tests/base.py
index 5b0cfe018..2eadc9a8c 100644
--- a/tests/base.py
+++ b/tests/base.py
@@ -2,6 +2,7 @@
 import pathlib
 import re
 import subprocess
+import tempfile
 import time
 
 import unittest