diff --git a/hosts/io/disko.nix b/hosts/io/disko.nix
index 37f3160..4e6c9d5 100644
--- a/hosts/io/disko.nix
+++ b/hosts/io/disko.nix
@@ -1,4 +1,8 @@
+{ inputs, ... }:
+
 {
+  imports = [ inputs.disko.nixosModules.default ];
+
   disko.devices.disk.main = {
     type = "disk";
     device = "/dev/disk/by-id/mmc-hDEaP3_0x1041b689";