Minor fix in the tcp_test.py