@@ -112,7 +112,10 @@ }; 
   services.colmenaAutoUpgrade = {
     enable = true;
-    pullGit = true;
+    git = {
+      enable = true;
+      branch = "origin/main";
+    };
     useNixShell = true;
     dates = "03:23";
     allowReboot = true;

@@ -63,12 +63,18 @@ default = false;       description = "Enable automatic upgrades for Colmena";
     };
 
-    pullGit = lib.mkOption {
-      default = false;
-      type = lib.types.bool;
-      description = ''
-        Whether to pull the latest changes from the Git repository before upgrading.
-      '';
+    git = lib.mkOption {
+      type = lib.types.submodule {
+        options = {
+          enable = lib.mkEnableOption "Whether to pull the latest changes from the Git repository before upgrading.";
+
+          branch = lib.mkOption {
+            type = lib.types.str;
+            default = "origin/main";
+            description = "Git branch to checkout after fetching";
+          };
+        };
+      };
     };
 
     useNixShell = lib.mkOption {
@@ -209,10 +215,10 @@ git = "${pkgs.gitMinimal}/bin/git";           nix-shell = "${pkgs.nix}/bin/nix-shell";
         in
         ''
-          ${lib.optionalString cfg.pullGit
+          ${lib.optionalString cfg.git.enable
             ''
-            ${git} fetch --all --prune
-            ${git} checkout FETCH_HEAD
+            ${git} fetch --prune
+            ${git} checkout ${cfg.git.branch}
             ''
           }
           ${if cfg.useNixShell then ''