Correct test to match new message