diff --git a/nixos/hosts/osaka-vultr-01/disks.nix b/nixos/hosts/osaka-vultr-01/disks.nix
index 66d6d76c..d7fd7a88 100644
--- a/nixos/hosts/osaka-vultr-01/disks.nix
+++ b/nixos/hosts/osaka-vultr-01/disks.nix
@@ -1,4 +1,6 @@
 {
+  boot.loader.grub.device = "/dev/vda";
+  boot.initrd.luks.devices."vda2".device = "/dev/vda2";
   disko.devices = {
     disk = {
       vda = {
diff --git a/nixos/small.nix b/nixos/small.nix
index b62e4252..e99fb941 100644
--- a/nixos/small.nix
+++ b/nixos/small.nix
@@ -1,5 +1,8 @@
 { lib, config, pkgs, hostname, stateVersion, username, desktop, gpu, inputs, platform, theme, ... }: {
   imports = [ 
+      # Modules
+      inputs.disko.nixosModules.disko
+      
       # Services
       ./common/services/openssh.nix
       ./common/services/fail2ban.nix