diff --git a/tests/backends/test_torch_native_compilation.py b/tests/backends/test_torch_native_compilation.py
index 2055941a1e92f25cdce6f6e62262543aef265e66..1868689a991b37f52e2c7a23f1cca63708298c1c 100644
--- a/tests/backends/test_torch_native_compilation.py
+++ b/tests/backends/test_torch_native_compilation.py
@@ -9,6 +9,7 @@ from os.path import dirname, isfile, join
 
 import pytest
 import sympy
+import tempfile
 
 import pystencils
 from pystencils_autodiff import create_backward_assignments