diff --git a/example/simple-efi.nix b/example/simple-efi.nix
index 1d12941..ecb3384 100644
--- a/example/simple-efi.nix
+++ b/example/simple-efi.nix
@@ -5,7 +5,7 @@
         device = builtins.elemAt disks 0;
         type = "disk";
         content = {
-          type = "table_gpt";
+          type = "gpt";
           partitions = {
             ESP = {
               type = "EF00";
diff --git a/tests/lib.nix b/tests/lib.nix
index 5a961c3..98d43ce 100644
--- a/tests/lib.nix
+++ b/tests/lib.nix
@@ -145,7 +145,7 @@
           machine.succeed("${nodes.machine.system.build.formatScript}")
           machine.succeed("${nodes.machine.system.build.mountScript}")
           machine.succeed("${nodes.machine.system.build.mountScript}") # verify that the command is idempotent
-          machine.succeed("${nodes.machine.system.build.disko}") # verify that we can destroy and recreate again
+          machine.succeed("${nodes.machine.system.build.diskoScript}") # verify that we can destroy and recreate again
         ''}
         ${lib.optionalString (testMode == "cli") ''
           # TODO use the disko cli here