Fix a problem where early legalization can cause token chain problems.