Update to support probe registrations