diff --git a/python/taichi/_testing.py b/python/taichi/_testing.py
index 9e3beb965a6c7..31b107eaa2f86 100644
--- a/python/taichi/_testing.py
+++ b/python/taichi/_testing.py
@@ -205,6 +205,5 @@ def wrapped(*args, **kwargs):
     'get_rel_eps',
     'approx',
     'allclose',
-    'make_temp_file',
     'test',
 ]
diff --git a/tests/python/test_gui.py b/tests/python/test_gui.py
index ee4683d012927..750e646b3824e 100644
--- a/tests/python/test_gui.py
+++ b/tests/python/test_gui.py
@@ -1,8 +1,8 @@
 import numpy as np
 import pytest
+from taichi._testing import make_temp_file
 
 import taichi as ti
-from taichi import make_temp_file
 
 
 @pytest.mark.parametrize('dtype', [ti.u8, ti.f32])
diff --git a/tests/python/test_image_io.py b/tests/python/test_image_io.py
index 7816fdca4e543..1f81e3605f7c0 100644
--- a/tests/python/test_image_io.py
+++ b/tests/python/test_image_io.py
@@ -2,9 +2,9 @@
 
 import numpy as np
 import pytest
+from taichi._testing import make_temp_file
 
 import taichi as ti
-from taichi import make_temp_file
 
 
 # jpg is also supported but hard to test here since it's lossy: