Fix the incorrect Tx / Rx data size issue