Set correct branch for badge