fix the message after button click