diff --git a/tools/ffx_update.sh b/tools/ffx_update.sh
index 7ab781c..2217c0a 100644
--- a/tools/ffx_update.sh
+++ b/tools/ffx_update.sh
@@ -2,7 +2,7 @@
 
 . ~/.local/share/ffx.venv/bin/activate
 pushd ~/.local/src/ffx/
-git checkout main
+git checkout "${1:-main}"
 git pull
 pip install --editable .
 popd