Fix branch mistake