1 files changed, 5 insertions, 3 deletions
diff --git a/telnet-server.js b/telnet-server.js
index 6cee554..479bd2b 100644
--- a/telnet-server.js
+++ b/telnet-server.js
@@ -22,7 +22,7 @@ class TelnetServer {
 
   async handleConnection(socket) {
     const interfacer = new TelnetInterfacer(socket)
-    const { appElement, renderInterval, cleanTerminal } = await setupClient({
+    const { appElement, cleanTerminal, flushable, renderInterval } = await setupClient({
       backend: this.backend,
       writable: socket,
       interfacer,
@@ -39,8 +39,9 @@ class TelnetServer {
       interfacer.cleanTelnetOptions()
       socket.write('\r' + msg + '\r\n')
       socket.end()
-      this.sockets.splice(this.sockets.indexOf(socket, 1))
+      flushable.end()
       closed = true
+      this.sockets.splice(this.sockets.indexOf(socket, 1))
     }
 
     appElement.on('quitRequested', quit)
@@ -48,8 +49,9 @@ class TelnetServer {
     socket.on('close', () => {
       if (!closed) {
         clearInterval(renderInterval)
-        this.sockets.splice(this.sockets.indexOf(socket, 1))
+        flushable.end()
         closed = true
+        this.sockets.splice(this.sockets.indexOf(socket, 1))
       }
     })