TCP mode bug fix