From b83bc79d91ae90da97ce0cf5b0bb2aae91859a1e Mon Sep 17 00:00:00 2001
From: codefiles <11915375+codefiles@users.noreply.github.com>
Date: Sat, 22 Feb 2025 18:20:06 -0500
Subject: [PATCH] Fix MBR conditional (#3192)

---
 archinstall/lib/disk/device_handler.py | 2 +-
 archinstall/lib/models/device_model.py | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/archinstall/lib/disk/device_handler.py b/archinstall/lib/disk/device_handler.py
index c099859b5c..3f476019ce 100644
--- a/archinstall/lib/disk/device_handler.py
+++ b/archinstall/lib/disk/device_handler.py
@@ -706,7 +706,7 @@ def partition(
 			if partition_table is None:
 				raise ValueError('Modification is marked as wipe but no partitioning table was provided')
 
-			if partition_table.MBR and len(modification.partitions) > 3:
+			if partition_table.is_mbr() and len(modification.partitions) > 3:
 				raise DiskError('Too many partitions on disk, MBR disks can only have 3 primary partitions')
 
 		# WARNING: the entire device will be wiped and all data lost
diff --git a/archinstall/lib/models/device_model.py b/archinstall/lib/models/device_model.py
index c3cc6df418..71eeff6f90 100644
--- a/archinstall/lib/models/device_model.py
+++ b/archinstall/lib/models/device_model.py
@@ -202,6 +202,9 @@ class PartitionTable(Enum):
 	GPT = 'gpt'
 	MBR = 'msdos'
 
+	def is_mbr(self) -> bool:
+		return self == PartitionTable.MBR
+
 
 class Units(Enum):
 	BINARY = 'binary'