Fixed editing message in logic