diff --git a/src/base/system/system.py b/src/base/system/system.py
index 999faa9..a7b112c 100644
--- a/src/base/system/system.py
+++ b/src/base/system/system.py
@@ -18,6 +18,8 @@ import psutil
 
 from base.scope import Scope
 from base.util.util import Util
+from base.system.disk_info import DiskInfo
+
 
 
 class System: