Fix branch checkout