diff --git a/flake.lock b/flake.lock
index 8527298..f095430 100644
--- a/flake.lock
+++ b/flake.lock
@@ -1,5 +1,25 @@
 {
   "nodes": {
+    "disko": {
+      "inputs": {
+        "nixpkgs": [
+          "nixpkgs"
+        ]
+      },
+      "locked": {
+        "lastModified": 1717097713,
+        "narHash": "sha256-M0tIapdiiB2piVTDK+aL7AnsjS656AY7l7htKN0dnQM=",
+        "owner": "nix-community",
+        "repo": "disko",
+        "rev": "9d5c673a6611b7bf448dbfb0843c75b9cce9cf1f",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-community",
+        "repo": "disko",
+        "type": "github"
+      }
+    },
     "flake-compat": {
       "flake": false,
       "locked": {
@@ -274,6 +294,7 @@
     },
     "root": {
       "inputs": {
+        "disko": "disko",
         "flake-utils": "flake-utils",
         "gomod2nix": "gomod2nix",
         "home-manager": "home-manager",
diff --git a/flake.nix b/flake.nix
index 85c33f9..b1bdcc5 100644
--- a/flake.nix
+++ b/flake.nix
@@ -29,6 +29,13 @@
       };
     };
 
+    disko = {
+      url = "github:nix-community/disko";
+      inputs = {
+        nixpkgs.follows = "nixpkgs";
+      };
+    };
+
     gomod2nix = {
       url = "github:nix-community/gomod2nix";
       inputs = {
@@ -181,6 +188,7 @@
             { nixpkgs.overlays = [ nur.overlay ]; }
             home-manager.nixosModules.home-manager
             inputs.sops-nix.nixosModules.sops
+            inputs.disko.nixosModules.disko
           ];
         })
         (import ./hosts inputs);