You need to replace this with `wait_for_notification_message_displayed` otherwise the test goes too quickly and doesn't see the message correctly.