Correct branch bug