ALU output stage, change logic slightly