Tweak branch code