diff --git a/aether/config.nix b/aether/config.nix
index 8cdfed0..b40a4a1 100644
--- a/aether/config.nix
+++ b/aether/config.nix
@@ -1,4 +1,4 @@
-{ config, lib, pkgs, self, ... }:
+{ config, lib, pkgs, aether, ... }:
 {
   networking.hostName = "toki-aether";
   time.timeZone = "America/New_York";
@@ -23,8 +23,8 @@
   # Aether modules
 
   imports = [
-    self.nixosModules.aether
-    self.nixosModules.deploy-rpi5
+    aether.aether
+    aether.deploy-rpi5
   ];
 
   aether.domain = "tokinanpa.dev";
diff --git a/flake.nix b/flake.nix
index 95c6cb0..edd65c2 100644
--- a/flake.nix
+++ b/flake.nix
@@ -37,8 +37,8 @@ outputs = inputs@{ self, nixpkgs, agenix, rpi5-kernel, ... }:
 
     nixosConfigurations."toki-aether" =
       nixpkgs.lib.nixosSystem {
+        specialArgs.aether = self.nixosModules;
         modules = [
-          { _module.args = inputs; }
           agenix.nixosModules.default
           ./aether/hardware-configuration.nix
           ./aether/config.nix