fixed some issues with the changes

1 files changed, 2 insertions, 5 deletions

diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index fbc11ca3..8e9e0234 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -390,12 +390,9 @@ class Filesystem():
 	# TODO:
 	#   When instance of a HDD is selected, check all usages and gracefully unmount them
 	#   as well as close any crypto handles.
-	def __init__(self, blockdevice):
+	def __init__(self, blockdevice,mode):
 		self.blockdevice = blockdevice
-		if hasUEFI():
-			self.mode = GPT
-		else:
-			self.mode = MBR
+		self.mode = mode
 
 	def __enter__(self, *args, **kwargs):
 		if self.blockdevice.keep_partitions is False: