fix branch.py