diff --git a/script/tests/tests/PS Tests/power-supply-A/power-supply-A.py b/script/tests/tests/PS Tests/power-supply-A/power-supply-A.py
index faa331f..2981542 100644
--- a/script/tests/tests/PS Tests/power-supply-A/power-supply-A.py	
+++ b/script/tests/tests/PS Tests/power-supply-A/power-supply-A.py	
@@ -1,7 +1,7 @@
 #########################################
 ###### DO NOT MODIFY THE CODE BELOW #####
 #########################################
-global print_log, sendFeedback, inspect
+global print_log, sendFeedback,  sys, inspect, os, traceback
 import sys, inspect, os, traceback
 
 def print_log(testName, DEVICE, text):
diff --git a/script/tests/tests/PS Tests/power-supply-B/power-supply-B.py b/script/tests/tests/PS Tests/power-supply-B/power-supply-B.py
index 3c8fb6f..946b473 100644
--- a/script/tests/tests/PS Tests/power-supply-B/power-supply-B.py	
+++ b/script/tests/tests/PS Tests/power-supply-B/power-supply-B.py	
@@ -1,6 +1,6 @@
 
 ###### Init - DO NOT MODIFY THE CODE BELOW ######
-global print_log, sendFeedback, inspect
+global print_log, sendFeedback,  sys, inspect, os, traceback
 import sys, inspect, os, traceback
 
 def print_log(testName, DEVICE, text):