diff --git a/src/shared/protocol.h b/src/shared/protocol.h
index a9f6c20..27f8d8d 100644
--- a/src/shared/protocol.h
+++ b/src/shared/protocol.h
@@ -5,6 +5,8 @@
 #include <stdbool.h>
 #include <stddef.h>
 
+typedef struct ssl_st SSL;
+
 typedef int Status;
 enum NET_STATUS { STATUS_OK, STATUS_ERROR, STATUS_FINISHED, STATUS_NEXT, STATUS_CHUNK, STATUS_MANIFEST, STATUS_CHECK };
 
diff --git a/src/shared/transport_tcp.c b/src/shared/transport_tcp.c
index 82e33ba..2050ea2 100644
--- a/src/shared/transport_tcp.c
+++ b/src/shared/transport_tcp.c
@@ -1,5 +1,6 @@
 #include "transport_tcp.h"
 #include "log.h"
+#include "protocol.h"
 #include <arpa/inet.h>
 #include <openssl/ssl.h>
 #include <signal.h>
@@ -152,6 +153,7 @@ void client_disconnect(Client *client) {
     SSL_shutdown(client->ssl);
     SSL_free(client->ssl);
     client->ssl = NULL;
+    io_set_ssl(NULL);
   }
   close(client->file_descriptor);
   if (client->ssh_child_pid > 0) {