1 files changed, 14 insertions, 6 deletions

diff --git a/flake.nix b/flake.nix
index 6d5f636..cd44ed5 100644
--- a/flake.nix
+++ b/flake.nix
@@ -45,18 +45,25 @@
         system: nixpkgs-stable.legacyPackages.${system}.nixfmt-tree
       );
 
-      apps = nixpkgs-stable.lib.genAttrs systems (system:
+      apps = nixpkgs-stable.lib.genAttrs systems (
+        system:
         let
           pkgs = nixpkgs-stable.legacyPackages.${system};
+          ruby = pkgs.ruby.withPackages (ps: [
+            pkgs.rubyPackages.tty-command
+          ]);
 
-          nixos-update = pkgs.writers.writeRubyBin "nixos-update" {
-            libraries = [ pkgs.rubyPackages.tty-command ];
+          nixos-update = pkgs.writeShellApplication  {
+            name = "nixos-update";
             runtimeInputs = [
               pkgs.sops
               pkgs.nixos-anywhere
-              pkgs.nixos-rebuild
+              pkgs.nixos-rebuild-ng
             ];
-          } (builtins.readFile ./nixos-update.rb);
+            text = ''
+              exec "${ruby}/bin/ruby" "${./nixos-update.rb}" "$@"
+            '';
+          };
 
         in
         {
@@ -64,6 +71,7 @@
             type = "app";
             program = "${nixos-update}/bin/nixos-update";
           };
-        });
+        }
+      );
     };
 }