diff --git a/.modules/base/proxmox-vm-hardware.nix b/.modules/base/proxmox-vm-hardware.nix
index 49e76b8..bd67ae9 100644
--- a/.modules/base/proxmox-vm-hardware.nix
+++ b/.modules/base/proxmox-vm-hardware.nix
@@ -12,13 +12,17 @@
     ];
 
   system.stateVersion = "22.11";
-  boot.loader.systemd-boot.enable = "true";
-
-  boot.initrd.availableKernelModules = [ "uhci_hcd" "ehci_pci" "ahci" "virtio_pci" "virtio_blk" ];
-  boot.initrd.kernelModules = [ ];
-  boot.kernelModules = [ "kvm-amd" ];
-  boot.extraModulePackages = [ ];
-
+  
+  boot = {
+    loader.systemd-boot.enable = "true";
+    initrd = {
+      availableKernelModules = [ "uhci_hcd" "ehci_pci" "ahci" "virtio_pci" "virtio_blk" "uas"];
+      kernelModules = [ ];
+    }
+    kernelModules = [ "kvm-amd" ];
+    extraModulePackages = [ ];
+  }
+  
   fileSystems."/" =
     { device = "/dev/disk/by-uuid/40fe3178-6ec1-450f-93fd-c359f2f3daf9";
       fsType = "ext4";
@@ -54,7 +58,6 @@
     };
 
     loader.timeout = 0;
-    initrd.availableKernelModules = [ "uas" "virtio_blk" "virtio_pci" ];
   };
 
 #  fileSystems."/" = {