Add messages to tests