diff --git a/flake.lock b/flake.lock
index 5f2fdba..1369e9b 100644
--- a/flake.lock
+++ b/flake.lock
@@ -1,37 +1,6 @@
 {
   "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": 1710631334,
         "narHash": "sha256-rL5LSYd85kplL5othxK5lmAtjyMOBg390sGBTb3LRMM=",
@@ -49,8 +18,7 @@
     },
     "root": {
       "inputs": {
-        "nixos-needsreboot": "nixos-needsreboot",
-        "nixpkgs": "nixpkgs_2"
+        "nixpkgs": "nixpkgs"
       }
     }
   },
diff --git a/ops/default.nix b/ops/default.nix
index 30b681e..3540522 100644
--- a/ops/default.nix
+++ b/ops/default.nix
@@ -5,6 +5,6 @@
     bitwarden-cli
     mkcert
     vault
-    nixos-generate
+    nixos-generators
   ];
 }