diff --git a/tests/test_cache_executor_mpi.py b/tests/test_cache_executor_mpi.py
index bc5e1226..0b8a657b 100644
--- a/tests/test_cache_executor_mpi.py
+++ b/tests/test_cache_executor_mpi.py
@@ -7,7 +7,7 @@
 
 
 try:
-    from executorlib import FileExecutor
+    from executorlib.cache.executor import FileExecutor
 
     skip_h5py_test = False
 except ImportError:
diff --git a/tests/test_cache_executor_serial.py b/tests/test_cache_executor_serial.py
index 96aa2df0..f0cffe9d 100644
--- a/tests/test_cache_executor_serial.py
+++ b/tests/test_cache_executor_serial.py
@@ -11,7 +11,7 @@
 from executorlib.standalone.thread import RaisingThread
 
 try:
-    from executorlib import FileExecutor
+    from executorlib.cache.executor import FileExecutor
     from executorlib.cache.shared import execute_tasks_h5
 
     skip_h5py_test = False