Fix branch coloring