diff --git a/src/main.rs b/src/main.rs
index 8dae014..9f4970d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -18,12 +18,12 @@ fn format_drive(drive_name: &str) -> rsfdisk::Result<()> {
 
     disk.partition_table_create(PartitionTableKind::GPT)?;
 
-    let boot_part_type = PartitionKind::builder()
-        .guid(Guid::EfiSystem)
+    let partition_type = PartitionKind::builder()
+        .guid(Guid::LinuxRootx86_64)
         .build()?;
 
     let boot = Partition::builder()
-        .boot_part_type(partition_type.clone())
+        .partition_type(partition_type.clone())
         .name("EFI")
         //Assuming 512 bytes per sector, 2_097_152 sectors <=> 1 GiB.
         .size_in_sectors(2_097_152)
@@ -31,12 +31,8 @@ fn format_drive(drive_name: &str) -> rsfdisk::Result<()> {
 
     let _ = disk.partition_add(boot)?;
 
-    let root_part_type = PartitionKind::builder()
-        .guid(Guid::LinuxRootx86_64)
-        .build()?;
-
     let root = Partition::builder()
-        .root_part_type(partition_type)
+        .partition_type(partition_type)
         .name("Root")
         .size_in_sectors(121_634_816)
         // 500GB