Track more staging branches for TE