diff --git a/archinstall/lib/models/device_model.py b/archinstall/lib/models/device_model.py
index 6d5ef0ae2e..72538f7958 100644
--- a/archinstall/lib/models/device_model.py
+++ b/archinstall/lib/models/device_model.py
@@ -6,6 +6,7 @@
 from enum import Enum
 from pathlib import Path
 from typing import NotRequired, TypedDict, override
+from uuid import UUID
 
 import parted
 from parted import Disk, Geometry, Partition
@@ -871,6 +872,8 @@ class PartitionModification:
 	partuuid: str | None = None
 	uuid: str | None = None
 
+	_obj_id: UUID | str = field(init=False)
+
 	def __post_init__(self) -> None:
 		# needed to use the object as a dictionary key due to hash func
 		if not hasattr(self, '_obj_id'):
@@ -1134,6 +1137,8 @@ class LvmVolume:
 	# mapper device path /dev/<vg>/<vol>
 	dev_path: Path | None = None
 
+	_obj_id: uuid.UUID | str = field(init=False)
+
 	def __post_init__(self) -> None:
 		# needed to use the object as a dictionary key due to hash func
 		if not hasattr(self, '_obj_id'):