Fix wait_on_rate_limit_notify