diff --git a/tests/test_native_tensorflow_compilation.py b/tests/test_native_tensorflow_compilation.py
index bb7e8c68ebf8e4b382dbbd8e1c23a02b8da05343..ad13bbb482f3a849053b8bcfadc59b79f6bfb4e5 100644
--- a/tests/test_native_tensorflow_compilation.py
+++ b/tests/test_native_tensorflow_compilation.py
@@ -9,20 +9,13 @@
 """
 
 import os
-import subprocess
-from os.path import join
-from sysconfig import get_paths
 
 import pytest
 import sympy
 
 import pystencils
-from pystencils.cpu.cpujit import get_compiler_config
-from pystencils.include import get_pystencils_include_path
 from pystencils_autodiff import create_backward_assignments
-from pystencils_autodiff._file_io import write_cached_content, write_file
 from pystencils_autodiff.backends.astnodes import TensorflowModule
-from pystencils_autodiff.tensorflow_jit import _compile_env
 
 
 def test_detect_cpu_vs_cpu():
@@ -45,11 +38,6 @@ def test_detect_cpu_vs_cpu():
 def test_native_tensorflow_compilation_cpu():
     tf = pytest.importorskip('tensorflow')
 
-    extra_flags = ['-I' + get_paths()['include'], '-I' + get_pystencils_include_path()]
-
-    compile_flags = tf.sysconfig.get_compile_flags()
-    link_flags = tf.sysconfig.get_link_flags()
-
     module_name = "Ololol"
 
     target = 'cpu'
@@ -86,11 +74,6 @@ def test_native_tensorflow_compilation_cpu():
 def test_native_tensorflow_compilation_gpu():
     tf = pytest.importorskip('tensorflow')
 
-    extra_flags = ['-I' + get_paths()['include'], '-I' + get_pystencils_include_path()]
-
-    compile_flags = tf.sysconfig.get_compile_flags()
-    link_flags = tf.sysconfig.get_link_flags()
-
     module_name = "Ololol"
 
     target = 'gpu'