Fixed the issue that the socket is closed too early (#127)

Fixed the issue that the socket is closed too early (#127) #133