Update origin HEAD when running git migration

This updates local reference to origin/HEAD, which is necessary for
new-branch to work as expected.

R=ehmaldonado@google.com

Change-Id: Ia620b3b01c0ad23bb4e969f9026fa6bcfdd88b90
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/tools/depot_tools/+/2730051
Commit-Queue: Josip Sokcevic <sokcevic@google.com>
Auto-Submit: Josip Sokcevic <sokcevic@google.com>
Reviewed-by: Edward Lesmes <ehmaldonado@chromium.org>
diff --git a/git_migrate_default_branch.py b/git_migrate_default_branch.py
index a605059..71be696 100644
--- a/git_migrate_default_branch.py
+++ b/git_migrate_default_branch.py
@@ -54,7 +54,10 @@
   if project_head != 'refs/heads/main':
     raise RuntimeError("The repository is not migrated yet.")
 
+  logging.info("Running fetch...")
   git_common.run('fetch', remote)
+  logging.info("Updating remote HEAD...")
+  git_common.run('remote', 'set-head', '-a', remote)
 
   branches = git_common.get_branches_info(True)