Fix message logic