fix a bug in chain_stage2.
2 files changed