Updated tests to look for new messages