Updated test assertions to match the actual messages