Add support to change the TCPSocket class