Fix checkout branch