TX_MUX.v: update MUX logic