diff --git a/flake.lock b/flake.lock
index 3f8987be..e2714182 100644
--- a/flake.lock
+++ b/flake.lock
@@ -7,15 +7,16 @@
         ]
       },
       "locked": {
-        "lastModified": 1694925805,
-        "narHash": "sha256-UNMivSc89undITtNoy6o6bf3Dck4v75rzGiMEXAPEB0=",
+        "lastModified": 1695379860,
+        "narHash": "sha256-ADsWLw33T/6APAoEfwPVkhuUHbH/BW/Jz5cgTgijsIs=",
         "owner": "nix-community",
         "repo": "disko",
-        "rev": "9ab96378f8cf602d5f3ce5a32f2c339509288d8e",
+        "rev": "26cf7576b85fd0b4070d8bd84475021e01d63814",
         "type": "github"
       },
       "original": {
         "owner": "nix-community",
+        "ref": "party",
         "repo": "disko",
         "type": "github"
       }
diff --git a/flake.nix b/flake.nix
index e9462e0f..6baad2b0 100644
--- a/flake.nix
+++ b/flake.nix
@@ -5,7 +5,7 @@
     nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
     floco.url = "github:aakropotkin/floco";
     floco.inputs.nixpkgs.follows = "nixpkgs";
-    disko.url = "github:nix-community/disko";
+    disko.url = "github:nix-community/disko/party";
     disko.inputs.nixpkgs.follows = "nixpkgs";
     sops-nix.url = "github:Mic92/sops-nix";
     sops-nix.inputs.nixpkgs.follows = "sops-nix";
diff --git a/pkgs/installer/flake-module.nix b/pkgs/installer/flake-module.nix
index 500a8551..b4a3564d 100644
--- a/pkgs/installer/flake-module.nix
+++ b/pkgs/installer/flake-module.nix
@@ -13,7 +13,7 @@ let
   };
 in
 {
-  flake.packages.x86_64-linux.install-iso = self.inputs.disko.lib.lib.makeDiskImage { nixosConfig = installer; };
+  flake.packages.x86_64-linux.install-iso = self.inputs.disko.lib.makeDiskImages { nixosConfig = installer; };
   flake.apps.x86_64-linux.install-vm.program = installer.config.formats.vm.outPath;
   flake.apps.x86_64-linux.install-vm-nogui.program = installer.config.formats.vm-nogui.outPath;
 }