diff --git a/tests/axi/test_axi.py b/tests/axi/test_axi.py
index e6a09d6..d288bdd 100644
--- a/tests/axi/test_axi.py
+++ b/tests/axi/test_axi.py
@@ -227,7 +227,7 @@ def test_axi(request, data_width):
         "sim_build_"+request.node.name.replace('[', '-').replace(']', ''))
 
     cocotb_test.simulator.run(
-        work_dir=tests_dir,
+        python_search=[tests_dir],
         verilog_sources=verilog_sources,
         toplevel=toplevel,
         module=module,
diff --git a/tests/axil/test_axil.py b/tests/axil/test_axil.py
index 13cef02..959e57d 100644
--- a/tests/axil/test_axil.py
+++ b/tests/axil/test_axil.py
@@ -204,7 +204,7 @@ def test_axil(request, data_width):
         "sim_build_"+request.node.name.replace('[', '-').replace(']', ''))
 
     cocotb_test.simulator.run(
-        work_dir=tests_dir,
+        python_search=[tests_dir],
         verilog_sources=verilog_sources,
         toplevel=toplevel,
         module=module,
diff --git a/tests/axis/test_axis.py b/tests/axis/test_axis.py
index e4d0174..2515cae 100644
--- a/tests/axis/test_axis.py
+++ b/tests/axis/test_axis.py
@@ -151,7 +151,7 @@ def test_axis(request, data_width):
         "sim_build_"+request.node.name.replace('[', '-').replace(']', ''))
 
     cocotb_test.simulator.run(
-        work_dir=tests_dir,
+        python_search=[tests_dir],
         verilog_sources=verilog_sources,
         toplevel=toplevel,
         module=module,