fix to use new discovery API