Fixed the message boxes