diff --git a/go-pool/stratum/stratum.go b/go-pool/stratum/stratum.go
index bc8a54c..89f3f09 100644
--- a/go-pool/stratum/stratum.go
+++ b/go-pool/stratum/stratum.go
@@ -186,10 +186,10 @@ func (s *StratumServer) handleClient(cs *Session, e *Endpoint) error {
 		data, isPrefix, err := connbuff.ReadLine()
 		if isPrefix {
 			log.Printf("Socket flood detected")
-			// TODO: Ban client
 			return errors.New("Socket flood")
 		} else if err == io.EOF {
-			log.Printf("Client disconnected")
+			log.Println("Client disconnected", cs.ip)
+			s.removeSession(cs)
 			break
 		} else if err != nil {
 			log.Printf("Error reading: %v", err)