diff --git a/src/AsyncTCP_TLS_Context.h b/src/AsyncTCP_TLS_Context.h
index 2f76fb1..e633bae 100644
--- a/src/AsyncTCP_TLS_Context.h
+++ b/src/AsyncTCP_TLS_Context.h
@@ -14,7 +14,8 @@
 #include "mbedtls/ctr_drbg.h"
 #include "mbedtls/error.h"
 
-#define ASYNCTCP_TLS_CAN_RETRY(r) (((r) == MBEDTLS_ERR_SSL_WANT_READ) || ((r) == MBEDTLS_ERR_SSL_WANT_WRITE))
+#define ASYNCTCP_TLS_CAN_RETRY(r)   (((r) == MBEDTLS_ERR_SSL_WANT_READ) || ((r) == MBEDTLS_ERR_SSL_WANT_WRITE))
+#define ASYNCTCP_TLS_EOF(r)         (((r) == MBEDTLS_ERR_SSL_CONN_EOF) || ((r) == MBEDTLS_ERR_SSL_PEER_CLOSE_NOTIFY))
 
 class AsyncTCP_TLS_Context
 {
@@ -39,8 +40,6 @@ private:
 
     int _socket;
 
-
-
     // Delete certificates used in handshake
     void _deleteHandshakeCerts(void);
 public: