diff --git a/tests/System/exe-script-with-input-single-location.py b/src/DIRAC/tests/Workflow/Integration/exe-script-with-input-single-location.py
similarity index 100%
rename from tests/System/exe-script-with-input-single-location.py
rename to src/DIRAC/tests/Workflow/Integration/exe-script-with-input-single-location.py
diff --git a/tests/System/exe-script-with-input.py b/src/DIRAC/tests/Workflow/Integration/exe-script-with-input.py
similarity index 100%
rename from tests/System/exe-script-with-input.py
rename to src/DIRAC/tests/Workflow/Integration/exe-script-with-input.py
diff --git a/tests/System/unitTestUserJobs.py b/tests/System/unitTestUserJobs.py
index 82c33bcef4d..6af9fa04dc5 100644
--- a/tests/System/unitTestUserJobs.py
+++ b/tests/System/unitTestUserJobs.py
@@ -11,6 +11,8 @@
 
 from DIRAC import gLogger
 from DIRAC.tests.Utilities.testJobDefinitions import *
+from DIRAC.Core.Security.ProxyInfo import getProxyInfo
+from DIRAC.DataManagementSystem.Client.DataManager import DataManager
 
 gLogger.setLevel("DEBUG")