diff --git a/archinstall/lib/disk.py b/archinstall/lib/disk.py
index 7d4a34c6..a8cba53b 100644
--- a/archinstall/lib/disk.py
+++ b/archinstall/lib/disk.py
@@ -342,6 +342,8 @@ class Filesystem():
 		else:
 			raise DiskError(f'The selected partition table format {self.mode} does not match that of {self.blockdevice}.')
 
+		return self
+
 	def __repr__(self):
 		return f"Filesystem(blockdevice={self.blockdevice}, mode={self.mode})"