Minor fix in the tcp_test