diff --git a/flake.lock b/flake.lock
index 5879aa4..d88b761 100644
--- a/flake.lock
+++ b/flake.lock
@@ -1,6 +1,37 @@
 {
   "nodes": {
+    "nixos-needsreboot": {
+      "inputs": {
+        "nixpkgs": "nixpkgs"
+      },
+      "locked": {
+        "lastModified": 1709963086,
+        "narHash": "sha256-EuwPKp8R1TOhJd7aEdEeE6if1LPIyzOFhvCh7FOCIfs=",
+        "rev": "f4c69fbc9bde2df34fe41662a0ec670bc44fb937",
+        "revCount": 12,
+        "type": "tarball",
+        "url": "https://api.flakehub.com/f/pinned/thefossguy/nixos-needsreboot/0.1.4/018e21be-e317-7c2c-bf25-826e32e95817/source.tar.gz"
+      },
+      "original": {
+        "type": "tarball",
+        "url": "https://flakehub.com/f/thefossguy/nixos-needsreboot/%2A.tar.gz"
+      }
+    },
     "nixpkgs": {
+      "locked": {
+        "lastModified": 1709884566,
+        "narHash": "sha256-NSYJg2sfdO/XS3L8XN/59Zhzn0dqWm7XtVnKI2mHq3w=",
+        "rev": "2be119add7b37dc535da2dd4cba68e2cf8d1517e",
+        "revCount": 556557,
+        "type": "tarball",
+        "url": "https://api.flakehub.com/f/pinned/NixOS/nixpkgs/0.2311.556557%2Brev-2be119add7b37dc535da2dd4cba68e2cf8d1517e/018e2117-9198-79f3-9df4-4997fc00989b/source.tar.gz"
+      },
+      "original": {
+        "type": "tarball",
+        "url": "https://flakehub.com/f/NixOS/nixpkgs/%2A.tar.gz"
+      }
+    },
+    "nixpkgs_2": {
       "locked": {
         "lastModified": 1709961763,
         "narHash": "sha256-6H95HGJHhEZtyYA3rIQpvamMKAGoa8Yh2rFV29QnuGw=",
@@ -18,7 +49,8 @@
     },
     "root": {
       "inputs": {
-        "nixpkgs": "nixpkgs"
+        "nixos-needsreboot": "nixos-needsreboot",
+        "nixpkgs": "nixpkgs_2"
       }
     }
   },
diff --git a/flake.nix b/flake.nix
index 92c31cf..97528f0 100644
--- a/flake.nix
+++ b/flake.nix
@@ -5,8 +5,9 @@
     # configuration.nix. You can also use latter versions if you wish to
     # upgrade.
     nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
+    nixos-needsreboot.url = "https://flakehub.com/f/thefossguy/nixos-needsreboot/*.tar.gz";
   };
-  outputs = inputs@{ self, nixpkgs, ... }: {
+  outputs = inputs@{ self, nixpkgs, nixos-needsreboot, ... }: {
     # NOTE: 'nixos' is the default hostname set by the installer
     nixosConfigurations.valhalla = nixpkgs.lib.nixosSystem {
       # NOTE: Change this to aarch64-linux if you are on ARM