Improve consistency between TCP and UDP RX datapaths