More register stage consolidation