diff --git a/flake.lock b/flake.lock
index 88c2212..375d459 100644
--- a/flake.lock
+++ b/flake.lock
@@ -7,11 +7,11 @@
         ]
       },
       "locked": {
-        "lastModified": 1711934712,
-        "narHash": "sha256-sBDe+QmX/QohlnKeSEzrftcXyZL5FY09OMjZ59Rpyy4=",
+        "lastModified": 1712325259,
+        "narHash": "sha256-9PipXYv1Ek1oi0WrGZYtlvnjoYS20QdSLLpJ+nHEbjE=",
         "owner": "nix-community",
         "repo": "disko",
-        "rev": "611c9ea53250f7bb22286b3d26872280a0e608f9",
+        "rev": "fbbac1aee6eae60c32b47e95447c4a3ec59d9773",
         "type": "github"
       },
       "original": {
@@ -27,11 +27,11 @@
         ]
       },
       "locked": {
-        "lastModified": 1712071848,
-        "narHash": "sha256-OQ0tZXcl0Qea5Xg9m4cK9kj8rfKaDba/BUh4fLoftEc=",
+        "lastModified": 1712171473,
+        "narHash": "sha256-IRw8f54GsnooUSci9wZpAN7AK/XTgfqeuLbETFi+NH4=",
         "owner": "mgit-at",
         "repo": "nix-unify",
-        "rev": "5b92428955bb9abda1635b0f83bbf33d6d76325a",
+        "rev": "1899c0fd2f32edbfafb0031d961ad9e0b1ce1d53",
         "type": "github"
       },
       "original": {
@@ -43,11 +43,11 @@
     },
     "nixpkgs": {
       "locked": {
-        "lastModified": 1711703276,
-        "narHash": "sha256-iMUFArF0WCatKK6RzfUJknjem0H9m4KgorO/p3Dopkk=",
+        "lastModified": 1712163089,
+        "narHash": "sha256-Um+8kTIrC19vD4/lUCN9/cU9kcOsD1O1m+axJqQPyMM=",
         "owner": "nixos",
         "repo": "nixpkgs",
-        "rev": "d8fe5e6c92d0d190646fb9f1056741a229980089",
+        "rev": "fd281bd6b7d3e32ddfa399853946f782553163b5",
         "type": "github"
       },
       "original": {
diff --git a/modules/base-tools.nix b/modules/base-tools.nix
index eeb824a..c502ecb 100644
--- a/modules/base-tools.nix
+++ b/modules/base-tools.nix
@@ -4,6 +4,7 @@
   programs.mtr.enable = true;
 
   environment.systemPackages = with pkgs; [
+    nixos-needsreboot # for updating
     htop
     curl
     dstat
diff --git a/pkgs/nixos-needsreboot/default.nix b/pkgs/nixos-needsreboot/default.nix
new file mode 100644
index 0000000..bd77d42
--- /dev/null
+++ b/pkgs/nixos-needsreboot/default.nix
@@ -0,0 +1,26 @@
+{ lib
+, rustPlatform
+, fetchFromGitHub
+}:
+
+rustPlatform.buildRustPackage rec {
+  pname = "nixos-needsreboot";
+  version = "unstable-2024-03-06";
+
+  src = fetchFromGitHub {
+    owner = "srounce";
+    repo = "nixos-needsreboot";
+    rev = "9089c7343fc6fe4bb2f899e85686a77e64b33cd6";
+    hash = "sha256-ACaUD16GQQeDYb5JXBE56JjV7bidztPFsZpaSLrFP/U=";
+  };
+
+  cargoHash = "sha256-A2xAjyQlvsuSiHAlFkVeVNpIshLbiKfwqX5TGCuyGRc=";
+
+  meta = with lib; {
+    description = "Determine if you need to reboot your NixOS machine";
+    homepage = "https://github.com/srounce/nixos-needsreboot";
+    license = licenses.free; # FIXME: nix-init did not found a license
+    maintainers = with maintainers; [ mkg20001 ];
+    mainProgram = "nixos-needsreboot";
+  };
+}