diff --git a/sonic_platform_base/fan_drawer_base.py b/sonic_platform_base/fan_drawer_base.py
index 5d9f92397..0f56c8579 100644
--- a/sonic_platform_base/fan_drawer_base.py
+++ b/sonic_platform_base/fan_drawer_base.py
@@ -5,6 +5,7 @@
     to interact with a fan drawer module in SONiC
 """
 
+import sys
 from . import device_base
 
 
diff --git a/sonic_platform_base/psu_base.py b/sonic_platform_base/psu_base.py
index 98aebec8e..95ec516c4 100644
--- a/sonic_platform_base/psu_base.py
+++ b/sonic_platform_base/psu_base.py
@@ -5,6 +5,7 @@
     to interact with a power supply unit (PSU) in SONiC
 """
 
+import sys
 from . import device_base