Fix test branching