rename this `wait_for_notification` as it could be any kind of notification, not just a messae.