merged with upstreamHEADmaster

1 files changed, 5 insertions, 1 deletions

diff --git a/archinstall/__main__.py b/archinstall/__main__.py
index e125930f..69a79855 100644
--- a/archinstall/__main__.py
+++ b/archinstall/__main__.py
@@ -5,9 +5,13 @@ import pathlib
 # Load .git version before the builtin version
 if pathlib.Path('./archinstall/__init__.py').absolute().exists():
 	spec = importlib.util.spec_from_file_location("archinstall", "./archinstall/__init__.py")
+
+	if spec is None or spec.loader is None:
+		raise ValueError('Could not retrieve spec from file: archinstall/__init__.py')
+
 	archinstall = importlib.util.module_from_spec(spec)
 	sys.modules["archinstall"] = archinstall
-	spec.loader.exec_module(sys.modules["archinstall"])
+	spec.loader.exec_module(archinstall)
 else:
 	import archinstall