modify decode unit to keep register write address.