diff --git a/flake.nix b/flake.nix
index 2204722..527a41f 100644
--- a/flake.nix
+++ b/flake.nix
@@ -126,7 +126,6 @@
 
         modules = [
           disko.nixosModules.disko
-          ./hosts/kube01/disko-config.nix
           ./hosts/kube01
         ];
       };
diff --git a/hosts/kube01/default.nix b/hosts/kube01/default.nix
index a19be4d..53f1ac3 100644
--- a/hosts/kube01/default.nix
+++ b/hosts/kube01/default.nix
@@ -3,6 +3,8 @@
 {
   imports =
     [
+      ../../modules/disko/efi-full-btrfs.nix
+
       ../../modules/nix.nix
       ../../modules/network-server.nix
       ../../modules/locale.nix
diff --git a/hosts/kube01/disko-config.nix b/modules/disko/efi-full-btrfs.nix
similarity index 99%
rename from hosts/kube01/disko-config.nix
rename to modules/disko/efi-full-btrfs.nix
index 24ffe29..5dcb78c 100644
--- a/hosts/kube01/disko-config.nix
+++ b/modules/disko/efi-full-btrfs.nix
@@ -1,7 +1,7 @@
 {
   disko.devices = {
     disk = {
-      vdb = {
+      sda = {
         type = "disk";
         device = "/dev/sda";
         content = {