diff --git a/hosts/riv-01/disko.nix b/hosts/riv-01/disko.nix
index f66d7e7..8ceddb0 100644
--- a/hosts/riv-01/disko.nix
+++ b/hosts/riv-01/disko.nix
@@ -1,4 +1,12 @@
 # hosts/riv-01/disko.nix
+{
+  config,
+  lib,
+  pkgs,
+  pkgsUnstable,
+  inputs,
+  ...
+}:
 {
   disko.devices = {
     disk = {