diff --git a/flake.nix b/flake.nix
index 22ae0f9..ff43dd0 100644
--- a/flake.nix
+++ b/flake.nix
@@ -8,10 +8,10 @@
       url = "github:nix-community/home-manager/release-26.05";
       inputs.nixpkgs.follows = "nixpkgs";
     };
-    # disko = {
-    #   url = "github:nix-community/disko";
-    #   inputs.nixpkgs.follows = "nixpkgs";
-    # };
+    disko = {
+      url = "github:nix-community/disko";
+      inputs.nixpkgs.follows = "nixpkgs";
+    };
   };
 
   outputs = { self, nixpkgs, nixos-conf-editor, home-manager, ... } @ inputs:
@@ -147,7 +147,7 @@ docker = nixpkgs.lib.nixosSystem {
             ./hosts/linode/configuration.nix
             ./modules/hardware-configuration/vm/linode.nix
             ./hosts/linode/disko.nix
-            nixpkgs.disko
+            inputs.nixosModules.disko
 
             home-manager.nixosModules.home-manager {
               home-manager.useGlobalPkgs = true;