diff --git a/configs/nixos/ni/disko.nix b/configs/nixos/ni/disko.nix
index 395cbbba..836bf1a1 100644
--- a/configs/nixos/ni/disko.nix
+++ b/configs/nixos/ni/disko.nix
@@ -1,9 +1,9 @@
-{ config, lib, ... }:
+{ disk ? "/dev/nvme0n1", config, lib, ... }:
 
 {
   disko.devices = {
     disk.primary = {
-      device = "/dev/nvme0n1";
+      device = disk;
       type = "disk";
       content = {
         type = "gpt";