diff --git a/modules/generic.nix b/modules/generic.nix
index 2549943..0838600 100644
--- a/modules/generic.nix
+++ b/modules/generic.nix
@@ -1,4 +1,4 @@
-{ config, pkgs, ... }:
+{ config, pkgs, lib, ... }:
 let
   sshPubkeys = [
     "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIE+X4vceRi79FLwwyzFzxNvaQlolQFrpYn0N4bgdLLaI root@hardlyworking"
@@ -55,7 +55,7 @@ in
 
   system.autoUpgrade = {
     enable = true;
-    allowReboot = true;
+    allowReboot = lib.mkDefault true;
     dates = "04:00";
     randomizedDelaySec = "30min";
   };