diff --git a/modules/system_info.py b/modules/system_info.py
index f6bcf77..1cd99f7 100644
--- a/modules/system_info.py
+++ b/modules/system_info.py
@@ -1,8 +1,9 @@
 import os
 import psutil
-from modules.base_module import DisplayModule
+from modules.base_module import BaseModule
+
+class SystemInfoModule(BaseModule):
 
-class SystemInfoModule(DisplayModule):
     def get_display_text(self):
         lines = []