diff --git a/modules/nixos/presets/disko/ext4/default.nix b/modules/nixos/presets/disko/ext4/default.nix
index 8d5d63a..55c5c22 100644
--- a/modules/nixos/presets/disko/ext4/default.nix
+++ b/modules/nixos/presets/disko/ext4/default.nix
@@ -25,7 +25,7 @@ in {
     disko.devices = {
       disk = {
         "${cfg.device}" = {
-          inherit device;
+          inherit (cfg) device;
           type = "disk";
           content = {
             type = "gpt";
diff --git a/modules/nixos/presets/disko/ext4Swap/default.nix b/modules/nixos/presets/disko/ext4Swap/default.nix
index 6f50c71..89b23c4 100644
--- a/modules/nixos/presets/disko/ext4Swap/default.nix
+++ b/modules/nixos/presets/disko/ext4Swap/default.nix
@@ -26,7 +26,7 @@ in {
     disko.devices = {
       disk.${cfg.device} = {
         type = "disk";
-        inherit device;
+        inherit (cfg) device;
         content = {
           type = "gpt"; # Initialize the disk with a GPT partition table
           partitions = {