diff --git a/template/flake.nix b/template/flake.nix
index be91b82..19ff084 100644
--- a/template/flake.nix
+++ b/template/flake.nix
@@ -39,9 +39,9 @@
     apps.${system} = {
       install = {
         program = "${pkgs.writeShellScript "install" ''
-          sudo run github:nix-community/disko/${inputs.disko.rev} -- --mode disko ${./system/disko.nix}
-          sudo nixos-generate-config --no-filesystems --force --root /mnt --dir ./system
-          sudo nixos-install --no-root-password --root /mnt --flake ${self}#nixos
+          sudo run ${inputs.disko} -- --mode disko ${./system/disko.nix}
+          sudo nixos-generate-config --force --no-filesystems --root /mnt --dir ./system
+          sudo nixos-install --no-root-password --root /mnt --flake ${self.outputs.nixosConfigurations.nixos}
         ''}";
         type = "app";
       };