Fix message on connection close