diff --git a/tests/integration/test_execution_environment.py b/tests/integration/test_execution_environment.py
index 72e481dc6..cd9ce55ff 100644
--- a/tests/integration/test_execution_environment.py
+++ b/tests/integration/test_execution_environment.py
@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import os
 import shlex
 
 from pathlib import Path
diff --git a/tests/integration/test_execution_environment_image.py b/tests/integration/test_execution_environment_image.py
index 71a33d7f0..4cff8206d 100644
--- a/tests/integration/test_execution_environment_image.py
+++ b/tests/integration/test_execution_environment_image.py
@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import os
 import shlex
 
 from pathlib import Path
diff --git a/tests/integration/test_pass_environment_variable.py b/tests/integration/test_pass_environment_variable.py
index b232b4045..460826945 100644
--- a/tests/integration/test_pass_environment_variable.py
+++ b/tests/integration/test_pass_environment_variable.py
@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import os
 import shlex
 
 from pathlib import Path
diff --git a/tests/integration/test_set_environment_variable.py b/tests/integration/test_set_environment_variable.py
index dcd6a086f..347e29830 100644
--- a/tests/integration/test_set_environment_variable.py
+++ b/tests/integration/test_set_environment_variable.py
@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import os
 import shlex
 
 from pathlib import Path