Set the correct branch for the basic block