How to use `diskoConfigurations` in a flake about disko HOT 4 CLOSED

error: attribute 'disko' missing

       at /nix/store/jpjdaylsd0clkzr9cg4n43yg816hccca-disko/share/disko/default.nix:10:49:

            9|       # _file = toString input;
           10|       imports = lib.singleton { disko.devices = cfg.disko.devices; };
             |                                                 ^
           11|       options = {
       Did you mean disk?
(use '--show-trace' to show detailed location information)

diff --git a/disko-config.nix b/disko-config.nix
index 9b50ce5..806bca1 100644
--- a/disko-config.nix
+++ b/disko-config.nix
@@ -1,4 +1,4 @@
-{ disk, memory, ... }: {
+{ disk, memory, ... }: { disko.devices = {
   disk = {
     vda = {
       type = "disk";
@@ -8,7 +8,6 @@
         format = "gpt";
         partitions = [
           {
-            type = "partition";
             name = "ESP";
             start = "1MiB";
             end = "512MiB";
@@ -23,7 +22,6 @@
             };
           }
           {
-            type = "partition";
             name = "luks";
             start = "512MiB";
             end = "100%";
@@ -47,14 +45,12 @@
         # LVs are created in alphabetic order for some reason
         # Prefixing them with a letter like this ensures desired creation order
         a_swap = {
-          type = "lvm_lv";
           size = memory;
           content = {
             type = "swap";
           };
         };
         z_persist = {
-          type = "lvm_lv";
           size = "100%FREE";
           content = {
             type = "btrfs";
@@ -85,4 +81,4 @@
       ];
     };
   };
-}
+};}
diff --git a/flake.nix b/flake.nix
index b3e1737..8baf1e3 100644
--- a/flake.nix
+++ b/flake.nix
@@ -27,13 +27,12 @@
           inputs.disko.nixosModules.disko
           inputs.home-manager.nixosModules.home-manager
           inputs.impermanence.nixosModules.impermanence
+          # Disko
+          (import ./disko-config.nix {
+            disk = "/dev/vda";
+            memory = "8G";
+          })
           {
-            # Disko
-            disko.devices = import ./disko-config.nix {
-              disk = "/dev/vda";
-              memory = "8G";
-            };
-
             # Home Manager
             home-manager.useGlobalPkgs = true;
             home-manager.useUserPackages = true;