diff --git a/common/autoUpgrade.nix b/common/autoUpgrade.nix
index 90fa4e88..a3b7264a 100644
--- a/common/autoUpgrade.nix
+++ b/common/autoUpgrade.nix
@@ -1,10 +1,4 @@
-{
-  config,
-  lib,
-  ...
-}: {
-  environment.variables.FLAKE = lib.mkDefault "github:alyraffauf/nixcfg";
-
+{config, ...}: {
   system.autoUpgrade = {
     enable = true;
     allowReboot = true;
diff --git a/common/base.nix b/common/base.nix
index 4c0a2ad0..0691f2be 100644
--- a/common/base.nix
+++ b/common/base.nix
@@ -2,6 +2,7 @@
   boot.loader.systemd-boot.configurationLimit = lib.mkDefault 10;
   console.useXkbConfig = true;
   hardware.keyboard.qmk.enable = true;
+  environment.variables.FLAKE = lib.mkDefault "github:alyraffauf/nixcfg/24.05";
 
   programs = {
     dconf.enable = true; # Needed for home-manager