diff --git a/flake.nix b/flake.nix
index 97725dd..c6e45a0 100644
--- a/flake.nix
+++ b/flake.nix
@@ -152,7 +152,9 @@
           darwinConfigurations.lookingglass.config.home-manager.users."Noah.Masur".home;
       };
 
-      diskoConfigurations = { swan = import ./hosts/swan/disks.nix; };
+      diskoConfigurations = {
+        swan = { ... }: (import ./hosts/swan/disks.nix { }).disko.devices;
+      };
 
       # Package servers into images with a generator
       packages = forAllSystems (system: {