diff --git a/disko/btrfs-legacy.nix b/disko/btrfs-legacy.nix
index e3ad455..a2880bc 100644
--- a/disko/btrfs-legacy.nix
+++ b/disko/btrfs-legacy.nix
@@ -1,4 +1,4 @@
-{ mainDisk }:
+{ mainDisk, ... }:
 
 {
     disk.main = {
diff --git a/flake.nix b/flake.nix
index abb1f56..cd4699f 100644
--- a/flake.nix
+++ b/flake.nix
@@ -16,7 +16,7 @@
         nixosConfigurations = {
             isengard = nixpkgs.lib.nixosSystem {
                 inherit system;
-                specialArgs = { inherit release mainDisk myUser; };
+                specialArgs = { inherit release mainDisk myUser disko; };
                 modules = [
                     ./hosts/isengard.nix
                     ./modules/base.nix