diff --git a/src/piler-smtp.c b/src/piler-smtp.c
index bbaeb46..e4db087 100644
--- a/src/piler-smtp.c
+++ b/src/piler-smtp.c
@@ -299,7 +299,9 @@
                }
             }
 
-            if(done){
+            /* Don't wait until the remote client closes the connection after he sent the QUIT command */
+
+            if(done || session->protocol_state == SMTP_STATE_FINISHED){
                tear_down_session(sessions, session->slot, &num_connections);
             }
          }
diff --git a/src/session.c b/src/session.c
index dda703a..fefa9ad 100644
--- a/src/session.c
+++ b/src/session.c
@@ -1,6 +1,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <netdb.h>
 #include <piler.h>
 
 
@@ -100,6 +101,10 @@
 
 
 void init_smtp_session(struct smtp_session *session, int slot, int sd, struct __config *cfg){
+   struct sockaddr_in addr;
+   socklen_t addr_size = sizeof(struct sockaddr_in);
+   char hbuf[NI_MAXHOST], sbuf[NI_MAXSERV];
+
    session->slot = slot;
 
    session->socket = sd;
@@ -119,6 +124,11 @@
    reset_bdat_counters(session);
 
    time(&(session->lasttime));
+
+   if(getpeername(sd, (struct sockaddr *)&addr, &addr_size) == 0 &&
+      getnameinfo((struct sockaddr *)&addr, addr_size, hbuf, sizeof(hbuf), sbuf, sizeof(sbuf), NI_NUMERICHOST | NI_NUMERICSERV) == 0){
+         snprintf(session->remote_host, INET6_ADDRSTRLEN-1, "%s", hbuf);
+   }
 }
 
 
@@ -211,4 +221,3 @@
    }
 
 }
-