diff --git a/flake.nix b/flake.nix
index e710ce59..4428baba 100644
--- a/flake.nix
+++ b/flake.nix
@@ -103,6 +103,7 @@
             modules = [
               ./hosts/${host}
               inputs.agenix.nixosModules.default
+              inputs.disko.nixosModules.disko
               inputs.home-manager.nixosModules.home-manager
               self.nixosModules.default
             ];
diff --git a/hosts/fallarbor/disko.nix b/hosts/fallarbor/disko.nix
index 90076db9..fb8e8e3a 100644
--- a/hosts/fallarbor/disko.nix
+++ b/hosts/fallarbor/disko.nix
@@ -1,7 +1,4 @@
-{inputs, ...}: {
-  imports = [
-    inputs.disko.nixosModules.disko
-  ];
+{
   disko.devices = {
     disk = {
       vdb = {
diff --git a/hosts/lavaridge/disko.nix b/hosts/lavaridge/disko.nix
index 47ce23df..afa3b8ff 100644
--- a/hosts/lavaridge/disko.nix
+++ b/hosts/lavaridge/disko.nix
@@ -1,7 +1,4 @@
-{inputs, ...}: {
-  imports = [
-    inputs.disko.nixosModules.disko
-  ];
+{
   disko.devices = {
     disk = {
       vdb = {
diff --git a/hosts/mandarin/disko.nix b/hosts/mandarin/disko.nix
index 47ce23df..afa3b8ff 100644
--- a/hosts/mandarin/disko.nix
+++ b/hosts/mandarin/disko.nix
@@ -1,7 +1,4 @@
-{inputs, ...}: {
-  imports = [
-    inputs.disko.nixosModules.disko
-  ];
+{
   disko.devices = {
     disk = {
       vdb = {
diff --git a/hosts/petalburg/disko.nix b/hosts/petalburg/disko.nix
index 47ce23df..afa3b8ff 100644
--- a/hosts/petalburg/disko.nix
+++ b/hosts/petalburg/disko.nix
@@ -1,7 +1,4 @@
-{inputs, ...}: {
-  imports = [
-    inputs.disko.nixosModules.disko
-  ];
+{
   disko.devices = {
     disk = {
       vdb = {
diff --git a/hosts/rustboro/disko.nix b/hosts/rustboro/disko.nix
index 44e593a0..d3ba7d57 100644
--- a/hosts/rustboro/disko.nix
+++ b/hosts/rustboro/disko.nix
@@ -1,7 +1,4 @@
-{inputs, ...}: {
-  imports = [
-    inputs.disko.nixosModules.disko
-  ];
+{
   disko.devices = {
     disk = {
       vdb = {