diff --git a/tests/test_container.py b/tests/test_container.py
index 3bdcc363..2ba9bb3a 100644
--- a/tests/test_container.py
+++ b/tests/test_container.py
@@ -1,6 +1,6 @@
 import pytest
 from unittest.mock import patch, MagicMock
-from executor import P
+from cgatcore.pipeline.execution import P
 
 
 @pytest.mark.parametrize("container_runtime, image, volumes, env_vars, expected_command", [