2-stage now passes asm tests