Fix discovery test