Removed hardcoded partition numbers when acessing partitions. As the order is not fixed and more options to disk layouts have been added.

1 files changed, 2 insertions, 2 deletions

diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index a8cba53b..73d913d0 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -354,9 +354,9 @@ class Filesystem():
 		b''.join(sys_command(f'sync'))
 		return True
 
-	def find_root_partition(self):
+	def find_partition(self, mountpoint):
 		for partition in self.blockdevice:
-			if partition.target_mountpoint == '/' or partition.mountpoint == '/':
+			if partition.target_mountpoint == mountpoint or partition.mountpoint == mountpoint:
 				return partition
 
 	def raw_parted(self, string:str):