diff --git a/cli/shell.nix b/cli/shell.nix
index 03add7f..abe2c57 100644
--- a/cli/shell.nix
+++ b/cli/shell.nix
@@ -12,23 +12,6 @@ let
   # use a more verbose output style for nix than default
   # only lix supports log-format = multiline-with-logs
   logFormat = if config.nixowos.system.lix.enable then "multiline-with-logs" else "bar-with-logs";
-
-  nixos-rebuild-diff = pkgs.writeShellApplication {
-    name = "nixos-rebuild-diff";
-
-    runtimeInputs = [ config.nix.package ];
-
-    text = ''
-      if [[ $* =~ build|boot|test|switch ]]; then
-        priorSystem=$(readlink /run/current-system)
-        newSystem=$(command nixos-rebuild --ask-sudo-password --log-format ${logFormat} "$@" | tail -n 1)
-
-        nix store diff-closures "$priorSystem" "$newSystem"
-      else
-        command nixos-rebuild --ask-sudo-password --log-format ${logFormat} "$@"
-      fi
-    '';
-  };
 in
 {
   options.nixowos.cli.${name} = {
@@ -47,7 +30,7 @@ in
       # make nix-shell preserve the user's $SHELL
       nix-shell = /* sh */ ''nix-shell --command "export SHELL=$SHELL; $SHELL"'';
 
-      nixos-rebuild = /* sh */ "${nixos-rebuild-diff}/bin/nixos-rebuild-diff";
+      nixos-rebuild = /* sh */ "nixos-rebuild --ask-sudo-password --log-format ${logFormat}";
 
       nix = /* sh */ "nix --log-format ${logFormat}";
     };