diff --git a/sd_webui/safe.py b/sd_webui/safe.py
index 276178c..92c5acf 100644
--- a/sd_webui/safe.py
+++ b/sd_webui/safe.py
@@ -7,4 +7,4 @@ if TYPE_CHECKING:
 
     unsafe_torch_load = torch.load
 else:
-    from module.safe import *
+    from modules.safe import *
diff --git a/sd_webui/shared.py b/sd_webui/shared.py
index 914a058..eaf0ea0 100644
--- a/sd_webui/shared.py
+++ b/sd_webui/shared.py
@@ -39,4 +39,4 @@ if TYPE_CHECKING:
     cmd_opts = argparse.Namespace()
 
 else:
-    from module.shared import *
+    from modules.shared import *