diff --git a/hosts/Legion/configuration.nix b/hosts/Legion/configuration.nix
index 81d5ae40..29dba241 100644
--- a/hosts/Legion/configuration.nix
+++ b/hosts/Legion/configuration.nix
@@ -41,7 +41,7 @@
   };
   system.autoUpgrade = {
     enable = true;
-    flake = "git+https://git.xenia.me.uk/xenia/nixos.git";
+    flake = "git+https://git.xenia.me.uk/xenia/nixos.git?ref=main";
     dates = "02:00";
     allowReboot = true;
     rebootWindow = {
diff --git a/hosts/Vanguard/configuration.nix b/hosts/Vanguard/configuration.nix
index dbcce2eb..58bfb04e 100644
--- a/hosts/Vanguard/configuration.nix
+++ b/hosts/Vanguard/configuration.nix
@@ -41,7 +41,7 @@
   };
   system.autoUpgrade = {
     enable = true;
-    flake = "git+https://git.xenia.me.uk/xenia/nixos.git";
+    flake = "git+https://git.xenia.me.uk/xenia/nixos.git?ref=main";
     allowReboot = false;
   };
   environment.systemPackages = with pkgs; [mesa];