selecting disk by id/size should return a BlockDevice()

1 files changed, 1 insertions, 1 deletions

diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index 2f3c122f..fe255991 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -278,4 +278,4 @@ def harddrive(size=None, model=None, fuzzy=False):
 		if model and (collection[drive]['model'] is None or collection[drive]['model'].lower() != model.lower()):
 			continue
 
-		return drive
\ No newline at end of file
+		return BlockDevice(drive['path'], drive)
\ No newline at end of file